머신러닝(9) confusionmatrix 값 시각화 하기

confusionmatrix 값을 그냥 가져가서 다른 사람들에게 보여주면 시각성이 매우 떨어지므로 시각화 작업이 필요하다. 이때 쓰는 함수가 히트맵이며 다음과 같이 쓸 수 있다.

 

# 8번에서 했던 confusionmatrix값을 기준으로 잡았다.

confusion_matrix(y_test, y_pred)
>>> array([[42,  6],
           [ 8, 24]], dtype=int64)

# cm값으로 저장
cm=confusion_matrix(y_test, y_pred)

#cm값 히트맵 작성
import seaborn as sb
sb.heatmap(data=cm,annot= True, cmap='RdPu',linewidths= 0.7)
plt.show()

 

그럼 이렇게 히트맵으로 시각성이 향상된 confusionmatrix값을 보여줄수 있다.