Mercurial > hg > nsaunier > traffic-intelligence
view trafficintelligence/tests/ml.txt @ 1139:3180a3bb63fe Etienne-Beauchamp/added-d-optional-argument-1581358671588
Close branch Etienne-Beauchamp/added-d-optional-argument-1581358671588.
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Sun, 22 Mar 2020 19:11:01 +0000 |
| parents | aafbc0bab925 |
| children |
line wrap: on
line source
>>> from math import fabs >>> from numpy import ones >>> from trafficintelligence.ml import prototypeCluster >>> nTrajectories = 7 >>> similarityFunc = lambda x, y: 1.-fabs(x-y)/(nTrajectories-1) >>> similarities = -ones((nTrajectories, nTrajectories)) >>> prototypeIndices = prototypeCluster(range(nTrajectories), similarities, 1., similarityFunc, optimizeCentroid = True) # too large to be similar >>> len(prototypeIndices) == nTrajectories True >>> # could use lists to have a length
