From ba71caca04fbc684da9bc1f377b8f0ff8306664e Mon Sep 17 00:00:00 2001
From: Vadim Pisarevsky <vadim.pisarevsky@gmail.com>
Date: Mon, 14 Dec 2015 16:37:19 +0300
Subject: [PATCH] fixed digits.py sample

---
 samples/python2/digits.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/samples/python2/digits.py b/samples/python2/digits.py
index b1688a8a07..04397a1431 100755
--- 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):
-- 
GitLab