view trafficintelligence/tests/ml.txt @ 1130:18e2c8cb9b09

Merged in etienne_beauchamp/trafficintelligence/Etienne-Beauchamp/added-d-optional-argument-1581358671588 (pull request #19) Etienne Beauchamp/added d optional argument 1581358671588 Approved-by: Nicolas Saunier <[email protected]>
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Tue, 11 Feb 2020 03:47:05 +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