인프런 커뮤니티 질문&답변
작성한 질문수
predict_proba(X_test)
작성
·
535
0
roc_auc_score 를 사용하여 수치를 확인할 때
앞에서는 여태까지 predict_proba 를 뽑아내어서 그냥 확률을 넣어줬는데
왜 여기서는 [:1]형태로 1차원 리스트를 만들어서 넣어주나요?
답변 1
0
권 철민
지식공유자
안녕하십니까,
predict_proba()는 sample별로 0과 1이 될 확률 각각 2개를 뽑아줍니다. 그리고 잘 아시다 시피 roc_auc_score(y_target, preds)입력할 때는 1이 될 확률만 입력해 줍니다.
아마도 전에는 roc_auc_score() 할 때는 preds= lr_clf.predict_proba(X_test) 한 뒤 roc_auc_score( y_target, preds[:, 1]) 과 같이 한것 같고, 이번에 preds= lr_clf.predict_proba(X_test)[:, 1] 후 roc_auc_score(y_target, preds) 한 것 같습니다.
둘 다 동일한 인자값을 roc_auc_score()에 입력한 것입니다.
감사합니다.