Mercurial > hg > nsaunier > traffic-intelligence

annotate trafficintelligence/tests/ml.txt @ 1306:4bc0651d91f9 default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
bug corrected generating last velocity twice and saving it (not saved, duplicated at loading time
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Mon, 30 Mar 2026 15:31:34 -0400
parents aafbc0bab925
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
980
23f98ebb113f first tests for clustering algo
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
1 >>> from math import fabs
23f98ebb113f first tests for clustering algo
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
2 >>> from numpy import ones
1030
aafbc0bab925 moved method around to avoid cross-dependencies
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1028
diff changeset
3 >>> from trafficintelligence.ml import prototypeCluster
980
23f98ebb113f first tests for clustering algo
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
4
23f98ebb113f first tests for clustering algo
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
5 >>> nTrajectories = 7
23f98ebb113f first tests for clustering algo
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
6 >>> similarityFunc = lambda x, y: 1.-fabs(x-y)/(nTrajectories-1)
23f98ebb113f first tests for clustering algo
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
7 >>> similarities = -ones((nTrajectories, nTrajectories))
23f98ebb113f first tests for clustering algo
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
8 >>> prototypeIndices = prototypeCluster(range(nTrajectories), similarities, 1., similarityFunc, optimizeCentroid = True) # too large to be similar
23f98ebb113f first tests for clustering algo
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
9 >>> len(prototypeIndices) == nTrajectories
23f98ebb113f first tests for clustering algo
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
10 True
23f98ebb113f first tests for clustering algo
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
11 >>> # could use lists to have a length