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 혼동 행렬 결과

    신간 소식 구독하기
    뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.