Commit 47d8187b authored by Joe Minichino's avatar Joe Minichino

fixed outdated params setting

parent a4a2659d
...@@ -86,20 +86,15 @@ class KNearest(StatModel): ...@@ -86,20 +86,15 @@ class KNearest(StatModel):
class SVM(StatModel): class SVM(StatModel):
def __init__(self, C = 1, gamma = 0.5): 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 = 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): def train(self, samples, responses):
self.model = cv2.ml.SVM_create() 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) self.model.train(samples, cv2.ml.ROW_SAMPLE, responses)
""" or this """
#self.model.train(samples, params = self.params)
def predict(self, samples): def predict(self, samples):
return self.model.predict(samples)[1][0].ravel() return self.model.predict(samples)[1][0].ravel()
......
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