인프런 커뮤니티 질문&답변
작성한 질문수
퀴즈
56%나 틀려요. 한번 도전해보세요!
이진 분류 문제에서 ROC AUC 지표 사용 시, 모델 예측 결과로 주로 제출하는 것은 무엇일까요?
클래스 0 예측값
클래스 1 예측값
예측 클래스 (0 또는 1)
클래스 1 예측 확률값
답변 3
1
paul1142
질문자
저도 계속 돌려봤는데.. 제가 글쎄 train['환불금액'].fillna(0) 요걸 그냥 train 에 받았더라구요. train['환불금액'] 에 받았어야 했어요 요걸.. 수정 하니 잘 돌아가요.. ㅠ
1
안녕하세요, 저도 동일에러 발생해서 해결한터라 부족하지만 답변드리겠습니다.
아마도 랜덤포레스트 model.fit 함수 동작시 y_train 설정 데이터를 train['성별']로 해서 발생한 에러로 보입니다.
train['성별']이 아닌 작성자께서 pop으로 저장하신 target 변수로 입력할 경우 정상 동작할것으로 보입니다.
아래 참고차 코드 올려 드립니다.
===============
target = train.pop('성별')
cols = train.select_dtypes(exclude='O').columns.to_list()
from sklearn.ensemble import RandomForestClassifier
model = RandomForestClassifier()
model.fit(train[cols], target)
model.predict_proba(test[cols])
답변 감사합니다 🙂