Merge pull request #5798 from vpisarev:fixed_digits_py

This commit is contained in:
Vadim Pisarevsky 2015-12-14 19:51:07 +00:00
commit d366cb960d

View File

@ -99,7 +99,7 @@ class SVM(StatModel):
self.model.train(samples, cv2.ml.ROW_SAMPLE, responses)
def predict(self, samples):
return self.model.predict(samples)[1][0].ravel()
return self.model.predict(samples)[1].ravel()
def evaluate_model(model, digits, samples, labels):