Cross validation에서 Confusion Matrix을 Metric으로 사용하기

여기서 clf는 model, x는 학습 데이터, y는 x의 label

 

대신

from sklearn.model_selection import cross_val_predict
from sklearn.metrics import confusion_matrix
y_pred = cross_val_predict(clf, X, y, cv=5)
conf_mat = confusion_matrix(y, y_pred)

이 값을 시각화 하는 방법은 다음과 같다.

from sklearn.model_selection import cross_val_predict
from sklearn.metrics import confusion_matrix
from sklearn.metrics import ConfusionMatrixDisplay

y_pred = cross_val_predict(clf, X, y, cv=5)
conf_mat = confusion_matrix(y, y_pred)
disp = ConfusionMatrixDisplay(conf_mat, display_labels=clf.classes_)
fig, ax = plt.subplots(figsize=(10, 10))
disp.plot(values_format='d', ax=ax)
plt.title("Logistic Regression", fontsize=25);