Mercurial > hg > nsaunier > traffic-intelligence

comparison python/ml.py @ 501:c81cbd6953fb

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
update to classify speed to remove data at both ends
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Tue, 20 May 2014 15:18:55 -0400
parents 8bc632cb8344
children 727e3c529519
comparison
equal deleted inserted replaced
500:3b99d712bbee 501:c81cbd6953fb
22 self.params = dict(svm_type = svm_type, kernel_type = kernel_type, degree = degree, gamma = gamma, coef0 = coef0, Cvalue = Cvalue, nu = nu, p = p) 22 self.params = dict(svm_type = svm_type, kernel_type = kernel_type, degree = degree, gamma = gamma, coef0 = coef0, Cvalue = Cvalue, nu = nu, p = p)
23 23
24 def train(self, samples, responses): 24 def train(self, samples, responses):
25 self.model.train(samples, responses, params = self.params) 25 self.model.train(samples, responses, params = self.params)
26 26
27 def predict(self, sample): 27 def predict(self, samples):
28 return np.float32(self.model.predict(s)) 28 return np.float32([self.model.predict(s) for s in samples])
29 29
30 30
31 class Centroid: 31 class Centroid:
32 'Wrapper around instances to add a counter' 32 'Wrapper around instances to add a counter'
33 33