Commit d366cb96 authored by Vadim Pisarevsky's avatar Vadim Pisarevsky

Merge pull request #5798 from vpisarev:fixed_digits_py

parents 5ac1d832 ba71caca
......@@ -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):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment