1. 程式人生 > >【分類】KNN分類演算法之Python實現

【分類】KNN分類演算法之Python實現

KNN稱為K最近鄰。對於待分類資料,它先計算出與其最相近的K個的樣本,然後判斷這K個樣本中最多的類標籤,並將待分類資料標記為這個最多的類標籤。

python樣例程式碼:

import numpy as np
from sklearn.neighbors import KNeighborsClassifier as kNN

K = 3

train_mat = np.array([[1, 2, 3], [2, 3, 5], [55, 33, 66], [55, 33, 66]])
label = np.array([0, 0, 1, 1])
neigh = kNN(n_neighbors=K, algorithm='auto', weights='distance', n_jobs=1)
neigh.fit(train_mat, label)
test_mat = np.array([[2, 3, 4], [51, 32, 24]])
res = neigh.predict(test_mat)
print(res)

輸出:

[0  1]