fixed digits.py sample

9 years ago · ba71caca04
parent 4c1c0b108d
commit ba71caca04
1 changed files with 1 additions and 1 deletions
--- a/samples/python2/digits.py
+++ b/samples/python2/digits.py
@ -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):