Confusion_matrix 메소드 정의

ㅇ 모듈 불러오기

from sklearn import metrics

from sklearn.metrics import confusion_matrix, classification_report 

 

ㅇ 결과 예시 

ㅇ 메소드 정의

1 ##############################

def plot_confusion_matrix(model):

    y_pred = model.predict(X_test)

    cm = confusion_matrix(y_test, y_pred)

    conf_matrix = pd.DataFrame(data = cm,columns = ['Predicted:0','Predicted:1'], index = ['Actual:0','Actual:1'])

    sns.heatmap(conf_matrix, annot = True, fmt = 'd', cmap = ListedColormap(['lightskyblue']), cbar = False, 

                linewidths = 0.1, annot_kws = {'size':25})

    plt.xticks(fontsize = 20)

    plt.yticks(fontsize = 20)

    plt.show()