파이썬에서 혼동 행렬의 각 요소에 대한 인덱스 가져 오기

scikit-learnconfusion-matrix 그럼 각각의 인덱스를 얻을 수 없었기 때문에.
(혹시 할 수 있니?)

조건



예를 들어 고양이인지 여부를 결정하기 위해 1이 고양이이고 0이 고양이가 아니라고 가정합니다.preds 를 예측, labels 를 정답 데이터로 한다.

절차



numpy 행렬로 변환
import numpy as np

preds = [0., 1., 1., ..., 0., 1., 0.]  # 適当な0or1のリスト
labels = [0., 1., 1., ..., 0., 1., 0.]  # 適当な0or1のリスト

preds = np.array(preds)
labels = np.array(labels)

고양이 색인
one_ind_p = preds == 1
one_ind_l = labels == 1
zero_ind_p = np.logical_not(one_ind_p)  # boolの反転
zero_ind_l = np.logical_not(one_ind_l)
# array([False,  True,  True, ..., False,  True, False]) みたいなデータになる

혼동 행렬의 각 요소의 인덱스를 얻습니다.
tp = np.argwhere(one_ind_l & one_ind_p)
fp = np.argwhere(zero_ind_l & one_ind_p)
fn = np.argwhere(one_ind_l & zero_ind_p)
tn = np.argwhere(zero_ind_l & zero_ind_p)

appendix



혼동 행렬은 이것

ht tp // // 마을 s 흠뻑 st. rg / 어서 x. php? F%에 5% 80% 아 4

좋은 웹페이지 즐겨찾기