Mercurial > hg > nsaunier > traffic-intelligence

comparison python/ml.py @ 665:15e244d2a1b5

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
corrected bug with circular import for VideoFilenameAddable, moved to base module
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 20 May 2015 13:57:47 +0200
parents 3058e00887bc
children da1352b89d02
comparison
equal deleted inserted replaced
664:455f9b93819c 665:15e244d2a1b5
1 #! /usr/bin/env python 1 #! /usr/bin/env python
2 '''Libraries for machine learning algorithms''' 2 '''Libraries for machine learning algorithms'''
3 3
4 import numpy as np 4 import numpy as np
5 5
6 __metaclass__ = type
7 6
8 class Model(object): 7 class Model(object):
9 '''Abstract class for loading/saving model''' 8 '''Abstract class for loading/saving model'''
10 def load(self, fn): 9 def load(self, fn):
11 self.model.load(fn) 10 self.model.load(fn)
26 25
27 def predict(self, samples): 26 def predict(self, samples):
28 return np.float32([self.model.predict(s) for s in samples]) 27 return np.float32([self.model.predict(s) for s in samples])
29 28
30 29
31 class Centroid: 30 class Centroid(object):
32 'Wrapper around instances to add a counter' 31 'Wrapper around instances to add a counter'
33 32
34 def __init__(self, instance, nInstances = 1): 33 def __init__(self, instance, nInstances = 1):
35 self.instance = instance 34 self.instance = instance
36 self.nInstances = nInstances 35 self.nInstances = nInstances