Mercurial > hg > nsaunier > traffic-intelligence

comparison 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
comparison
equal deleted inserted replaced
1032:d0e339359d8a 1033:8ffb3ae9f3d2
208 if similarityFunc is None: 208 if similarityFunc is None:
209 print('similarityFunc is None') 209 print('similarityFunc is None')
210 return None 210 return None
211 211
212 # sort instances based on length 212 # sort instances based on length
213 indices = range(len(instances)) 213 indices = list(range(len(instances)))
214 if randomInitialization or optimizeCentroid: 214 if randomInitialization or optimizeCentroid:
215 indices = np.random.permutation(indices).tolist() 215 indices = np.random.permutation(indices).tolist()
216 else: 216 else:
217 def compare(i, j): 217 def compare(i, j):
218 if len(instances[i]) > len(instances[j]): 218 if len(instances[i]) > len(instances[j]):
219 return -1 219 return -1
220 elif len(instances[i]) == len(instances[j]): 220 elif len(instances[i]) == len(instances[j]):
221 return 0 221 return 0
222 else: 222 else:
223 return 1 223 return 1
224 indices.sort(compare) 224 indices.sort(key=lambda i: len(instances[i]))
225 # initialize clusters 225 # initialize clusters
226 clusters = [] 226 clusters = []
227 if initialPrototypeIndices is None: 227 if initialPrototypeIndices is None:
228 prototypeIndices = [indices[0]] 228 prototypeIndices = [indices[0]]
229 else: 229 else: