exposed parallelized SVM prediction to python (predict_all)

2012-06-14 10:02:49 +00:00
parent e4d9d5294e
commit a98d6b6217
3 changed files with 9 additions and 2 deletions
--- a/samples/python2/letter_recog.py
+++ b/samples/python2/letter_recog.py
@@ -88,7 +88,7 @@ class SVM(LetterStatModel):
        self.model.train(samples, responses, params = params)

    def predict(self, samples):
-        return np.float32( [self.model.predict(s) for s in samples] )
+        return self.model.predict_all(samples).ravel()


 class MLP(LetterStatModel):