Classify.py

from sklearn.neighbors import KNeighborsClassifier X = [[0], [1], [2], [3], [4], [5], [6], [7], [8]]#data y = [0, 0, 0, 1, 1, 1, 2, 2, 2]#target neigh = KNeighborsClassifier(n_neighbors=3) neigh.fit(X, y)