fixed digits.py sample

This commit is contained in:
Vadim Pisarevsky 2015-12-14 16:37:19 +03:00
parent 4c1c0b108d
commit ba71caca04

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):