| 76 | self.model.save(fn) |
| 77 | |
| 78 | class KNearest(StatModel): |
| 79 | def __init__(self, k = 3): |
| 80 | self.k = k |
| 81 | self.model = cv2.ml.KNearest_create() |
| 82 | |
| 83 | def train(self, samples, responses): |
| 84 | self.model.train(samples, cv2.ml.ROW_SAMPLE, responses) |
| 85 | |
| 86 | def predict(self, samples): |
| 87 | retval, results, neigh_resp, dists = self.model.findNearest(samples, self.k) |
| 88 | return results.ravel() |
| 89 | |
| 90 | class SVM(StatModel): |
| 91 | def __init__(self, C = 1, gamma = 0.5): |