Mercurial > hg > nsaunier > traffic-intelligence
annotate python/tests/ml.txt @ 983:7463c9bc846b
work in progress on script to manage large dataset with multiple sites
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Tue, 27 Feb 2018 16:46:06 -0500 |
| parents | 23f98ebb113f |
| 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 |
|
23f98ebb113f
first tests for clustering algo
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
3 >>> from ml import prototypeCluster |
|
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 |
