Mercurial Hosting > traffic-intelligence

diff trafficintelligence/ml.py @ 1033:8ffb3ae9f3d2

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
work in progress
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 20 Jun 2018 00:07:03 -0400
parents cc5cb04b04b0
children 75a6ad604cc5
line wrap: on
line diff
--- a/trafficintelligence/ml.py	Tue Jun 19 17:07:50 2018 -0400
+++ b/trafficintelligence/ml.py	Wed Jun 20 00:07:03 2018 -0400
@@ -210,7 +210,7 @@
         return None
 
     # sort instances based on length
-    indices = range(len(instances))
+    indices = list(range(len(instances)))
     if randomInitialization or optimizeCentroid:
         indices = np.random.permutation(indices).tolist()
     else:
@@ -221,7 +221,7 @@
                 return 0
             else:
                 return 1
-        indices.sort(compare)
+        indices.sort(key=lambda i: len(instances[i]))
     # initialize clusters
     clusters = []
     if initialPrototypeIndices is None: