Merge pull request #5798 from vpisarev:fixed_digits_py
This commit is contained in:
commit
d366cb960d
@ -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):
|
||||
|
Loading…
Reference in New Issue
Block a user