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): ...@@ -77,7 +77,6 @@ class KNearest(StatModel):
self.model = cv2.ml.KNearest_create() self.model = cv2.ml.KNearest_create()
def train(self, samples, responses): def train(self, samples, responses):
self.model = cv2.ml.KNearest_create()
self.model.train(samples, cv2.ml.ROW_SAMPLE, responses) self.model.train(samples, cv2.ml.ROW_SAMPLE, responses)
def predict(self, samples): def predict(self, samples):
...@@ -93,7 +92,6 @@ class SVM(StatModel): ...@@ -93,7 +92,6 @@ class SVM(StatModel):
self.model.setType(cv2.ml.SVM_C_SVC) 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.train(samples, cv2.ml.ROW_SAMPLE, responses) self.model.train(samples, cv2.ml.ROW_SAMPLE, responses)
def predict(self, samples): 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