From 47d8187bc3c481a78938d89779f539e08b88bc99 Mon Sep 17 00:00:00 2001
From: Joe Minichino <joe.minichino@gmail.com>
Date: Thu, 28 May 2015 15:20:08 +0100
Subject: [PATCH] fixed outdated params setting
---
samples/python2/digits.py | 13 ++++---------
1 file changed, 4 insertions(+), 9 deletions(-)
diff --git a/samples/python2/digits.py b/samples/python2/digits.py
index 18cdd4ce75..a0ea337b70 100755
--- a/samples/python2/digits.py
+++ b/samples/python2/digits.py
@@ -86,20 +86,15 @@ class KNearest(StatModel):
class SVM(StatModel):
def __init__(self, C = 1, gamma = 0.5):
- self.params = dict( kernel_type = cv2.ml.SVM_RBF,
- svm_type = cv2.ml.SVM_C_SVC,
- C = C,
- gamma = gamma )
self.model = cv2.ml.SVM_create()
+ self.model.setGamma(gamma)
+ self.model.setC(C)
+ self.model.setKernel(cv2.ml.SVM_RBF)
+ self.model.setType(cv2.ml.SVM_C_SVC)
def train(self, samples, responses):
self.model = cv2.ml.SVM_create()
- """ original code """
- #self.model.train(samples, responses, params = self.params)
- """ but it's either this """
self.model.train(samples, cv2.ml.ROW_SAMPLE, responses)
- """ or this """
- #self.model.train(samples, params = self.params)
def predict(self, samples):
return self.model.predict(samples)[1][0].ravel()