Mercurial Hosting > traffic-intelligence
comparison python/ml.py @ 665:15e244d2a1b5
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 |