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