분류 문제에 나이브 베이즈 적용하기
나이브 베이즈 알고리즘으로 분류 문제를 풀어 봅시다.
1. GaussianNB() 함수로 모델을 훈련합니다.
▼ mynaivebayes.ipynb
[in :]
from sklearn.naive_bayes import GaussianNB
classifier = GaussianNB()
classifier.fit(X_train, y_train)
2. 훈련된 모델을 이용해 결과를 예측합니다. 테스트 데이터셋인 X_test의 라벨을 예측합니다.
[in :]
Predicting the Test set results
y_pred = classifier.predict(X_test)
cm = metrics.confusion_matrix(y_test, y_pred)
cm
3. 혼동 행렬을 출력합니다.
[out:]
array([[66, 2],
[ 6, 26]])