[텍스트 분석] 혼동 행렬

혼동행렬 만들기

from sklearn.metrics import confusion_matrix
confusion_matrix(y_test, y_pred)

 

지표

임포트

from sklearn.metrics import *

정확도

accuracy_score(y_test, y_pred)

정밀도

precision_score(y_test, y_pred)

재현도

recall_score(y_test, y_pred)

 

ROC 곡선

임포트

from sklearn.metrics import roc_auc_score, roc_curve
import matplotlib.pyplot as plt

시각화

fpr, tpr, threshold = roc_curve(y_test, prob)
plt.plot(fpr, tpr)

AUC

roc_auc_score(y_test, prob)