All thing of the world!

사이킷런(scikit-learn) KNeighborsClassifier 위주 간단 정리 본문

IT/인공지능(AI)

사이킷런(scikit-learn) KNeighborsClassifier 위주 간단 정리

WorldSeeker 2021. 5. 16. 21:49

사이킷런(scikit-learn) KNeighborsClassifier 사용 방법 위주로 정리합니다. 

 

▶ KNeighborsClassifier는 k-최근접 이웃 분류 모델을 만드는 사이킷런(scikit-learn) 클래스

 

Parameter

- n_neighbors : 이웃의 갯수를 지정 (기본값 5)

- p : 거래를 재는 방법, 1은 맨해튼 거리, 2는 - 유클리디안 거리(기본값 2)

- n_jobs : CPU 코어 지정 (기본값 1, -1은 모든 CPU 코어 사용)

 

Method

- fit() : 모델을 데이터로 훈련시킴

- predict() : 모델을 데이터로 훈련하고 예측함 (주어진 값을 가지고 모델이 판단) 

- score() : 훈련된 모델의 신뢰도 측정

- kneighbors() :  입력한 데이터에 가장 가까운 이웃 중 n_neighbors개의 거리와 인덱스를 반환

 

Function

- train_test_split() : 훈련 데이터셋과 테스트 데이터셋으로 나눔

 

Comments