Method predict

machine-learning/k-nearest-neighbors.py:34–40

(self, to_classify)

        self.y_train = training_labels

    def predict(self, to_classify):
        predictions = []
        for row in to_classify:
            label = self.closest(row)
            predictions.append(label)

        return predictions

iris = datasets.load_iris()