인프런 커뮤니티 질문&답변
작성한 질문수
작업형2 모의문제1 오류 문의
해결된 질문
작성
·
300
답변 1
0
퇴근후딴짓
지식공유자
분류에서 예측을 할 때 두 가지가 있어요
predict와 predict_proba가 있습니다.
predict은 클래스(종류)를 예측하고 이진 분류면 0과1, 다중분류면 0, 1, 2, 3, 등
predict_proba는 각 클래스의 확률값을 출력합니다. 이진분류면 0일 확률, 1일 확률
roc_auc는 1(양성)일 확률값을 통해 계산해요!
y_pred_proba = classifier.predict_proba(X_test)
roc_auc = roc_auc_score(y_test, y_pred_proba[:, 1])그외 f1, accuracy등 분류 평가 지표는 predict로 예측하면 됩니다.
y_pred = classifier.predict(X_test)
f1 = f1_score(y_test, y_pred)