더북(TheBook)

sklearn의 용어는 차치하고라도, sklearn 문서에서 모델은 어떤 데이터로 학습하고 또 다른 데이터를 예측하는 데 사용합니다. 훈련과 테스트 데이터를 분리했다면, 훈련 데이터로 모델을 학습하고 학습한 모델을 사용해서 테스트 데이터에 대한 예측을 수행합니다.

1. 3-NN 모델을 만듭니다.

2. 훈련 데이터로 모델을 적합합니다.

3. 그 모델을 사용해서 테스트 데이터에 대한 예측을 수행합니다.

4. 정확도를 사용해서 예측 결과를 평가합니다.

 

In [9]:

# 기본 옵션 n_neighbors = 5
knn = neighbors.KNeighborsClassifier(n_neighbors=3)
fit = knn.fit(iris_train_ftrs, iris_train_tgt)
preds = fit.predict(iris_test_ftrs)
# 예측 결과를 따로 분리해 둔 테스트 타깃으로 평가합니다

print("3NN accuracy:", metrics.accuracy_score(iris_test_tgt, preds))
3NN accuracy: 1.0
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.