본문 바로가기
인공지능/Machine Learning

[Machine Learning] 분류예측 K-Nearest Neighbor(K-NN)

by coding_su 2022. 12. 2.

📝머신러닝 K-Nearest Neighbor(K-NN)

from sklearn.neighbors import KNeighborsClassifier

 

KNeighborsClassifier() : 가장 가까운 n개 데이터(이웃데이터)로 분류

# 가장 가까운 n개 데이터로 분류
classifier = KNeighborsClassifier(n_neighbors= n)
# n_neighbors= 몇개의 데이터로 분석할건지 입력(이웃갯수) 디폴트값은 5

classifier = KNeighborsClassifier()

classifier.fit(X_train, y_train)
y_pred = classifier.predict(X_test)

from sklearn.metrics import confusion_matrix, accuracy_score

cm = confusion_matrix(y_test, y_pred)
accuracy_score(y_test, y_pred)

댓글