Mercurial Hosting > traffic-intelligence
comparison python/tests/ml.txt @ 980:23f98ebb113f
first tests for clustering algo
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Mon, 19 Feb 2018 16:32:59 -0500 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| 979:cc89267b5ff9 | 980:23f98ebb113f |
|---|---|
| 1 >>> from math import fabs | |
| 2 >>> from numpy import ones | |
| 3 >>> from ml import prototypeCluster | |
| 4 | |
| 5 >>> nTrajectories = 7 | |
| 6 >>> similarityFunc = lambda x, y: 1.-fabs(x-y)/(nTrajectories-1) | |
| 7 >>> similarities = -ones((nTrajectories, nTrajectories)) | |
| 8 >>> prototypeIndices = prototypeCluster(range(nTrajectories), similarities, 1., similarityFunc, optimizeCentroid = True) # too large to be similar | |
| 9 >>> len(prototypeIndices) == nTrajectories | |
| 10 True | |
| 11 >>> # could use lists to have a length |