Mercurial Hosting > traffic-intelligence
annotate trafficintelligence/tests/ml.txt @ 1154:2795d0e114c9
deal with possibility of prototype with speed 0 that crashes motion prediction
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 09 Sep 2020 16:28:38 -0400 |
parents | aafbc0bab925 |
children |
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 |