강의

멘토링

로드맵

인프런 커뮤니티 질문&답변

최은솔님의 프로필 이미지
최은솔

작성한 질문수

분류모델(RandomForestClassifier) 에서 y_train 값을 0 또는 1로 변환하는 것

작성

·

225

·

수정됨

0

y = (y_train['income']=='>50K').astype(int)

안녕하세요! 수업 잘 듣고 있습니다.

RandomForestClassifier 모델링할 때, target 인 y_train 값을 위와 같이 object 형에서 0 또는 1로 변환해서 모델링을 진행하잖아요~

근데 만약에 채점자는 '>50K' 를 1로 설정하지 않고 '<=50K' 를 1로 설정할 수도 있지 않을까해서요!

그럼 결과가 반대로 나올텐데...무엇을 1로 설정할지 명확히 문제에 주어지는건가요??

답변 1

0

안녕하세요 최은솔 님, 인프런 AI 인턴이에요.
분류모델(RandomForestClassifier)에서 target 값을 0 또는 1로 변환하는 것은 일반적인 방법 중 하나입니다. 하지만, 채점자가 ‘<=50K’를 1로 설정할 수 있는 경우 문제에 명시되어 있지 않다면, 이는 일반적으로는 예상되지 않는 상황입니다.
일반적인 문제 설정에서는 ‘>50K’가 1, ‘<=50K’가 0으로 설정되어 있고, 모델을 훈련하여 예측값을 구하게 됩니다. 채점자가 이를 다르게 설정할 수도 있는 상황이라면 문제의 명확한 지시사항이나 채점기준을 확인하는 것이 좋습니다. 감사합니다!

최은솔님의 프로필 이미지
최은솔

작성한 질문수

질문하기