Commit c4c3f1b2 authored by John Stowers's avatar John Stowers

fix digits.py to not ignore model parameters

otherwise model is only ever trained with default
parameters, and digits_adjust does nothing (for example)
parent 429d2805
......@@ -77,7 +77,6 @@ class KNearest(StatModel):
self.model = cv2.ml.KNearest_create()
def train(self, samples, responses):
self.model = cv2.ml.KNearest_create()
self.model.train(samples, cv2.ml.ROW_SAMPLE, responses)
def predict(self, samples):
......@@ -93,7 +92,6 @@ class SVM(StatModel):
self.model.setType(cv2.ml.SVM_C_SVC)
def train(self, samples, responses):
self.model = cv2.ml.SVM_create()
self.model.train(samples, cv2.ml.ROW_SAMPLE, responses)
def predict(self, samples):
......
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