혼동행렬(confusion matrix) 보는법

김우빈·2022년 6월 1일
0

JUST DEEP_HINT

목록 보기
5/6
post-thumbnail

#Confusion matrix 보는법

import sklearn
from sklearn.metrics import confusion_matrix
#scikit-learn을 이용해서 confusion matrix 도출
print(confusion_matrix(np.argmax(y_valid,axis=-1), np.argmax(pred,axis=-1)))

결과

[[ 977    0    0    0    0    1    1    1    0    0]
 [   0 1134    0    1    0    0    0    0    0    0]
 [   1    0 1028    0    0    0    0    3    0    0]
 [   0    0    1 1006    0    2    0    0    1    0]
 [   0    0    0    0  972    0    0    0    1    9]
 [   1    1    0    5    0  884    1    0    0    0]
 [   6    4    0    0    1    0  946    0    0    1]
 [   0    4    0    0    1    0    0 1022    0    1]
 [   1    0    0    1    1    1    0    0  966    4]
 [   0    2    0    0    4    2    0    2    1  998]]

오늘도 깊은 하루 되세요

profile
DeepLearning, MLOps

0개의 댓글