from sklearn.metrics import confusion_matrix import seaborn as sns y_test_pred = model.predict_classes(X_test) c_matrix = confusion_matrix(y_test, y_test_pred) ax = sns.heatmap(c_matrix, annot=True, xticklabels=['No Diabetes','Diabetes'], yticklabels=['No Diabetes','Diabetes'], cbar=False, cmap='Blues') ax.set_xlabel("Prediction") ax.set_ylabel("Actual") plt.show() plt.clf()
코드를 실행해 출력한 혼동 행렬은 그림 2-18과 같다.
▲ 그림 2-18 혼동 행렬 결과