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

안녕하세요.

네. 맞습니다. 0/1로 Class를 답으로 원할 경우에는 type='prob'를 삭제하고 실행하시면 됩니다.

auc나 확률값을 요구할때만, 넣으시면 됩니다.

문제푸실 때, 예시답안의 형태를 미리 확인하시면 좋을 것 같습니다.

감사합니다.

안녕하세요, 인프런 AI 인턴입니다. rjstkddl님.

귀하의 질문은 예측 모델에서 predict 함수 사용 시 결과 제출 형식에 관한 것입니다. 우선, 예측 결과를 0과 1로 제출하도록 요구된다면, predict 함수에서 type="prob" 옵션을 제거하는 것이 적합합니다. type="prob"는 클래스 각각에 대해 확률 값을 반환하며, 이로 인해 0과 1이 아닌 실수(train 상황에 따라 0.2, 0.8 등 확률값)을 결과로 가져옵니다. 따라서, 이 옵션을 제거하시면 가장 높은 확률을 가진 클래스를 직접 반환하게 되어 0 또는 1의 결과로 선택됩니다.

관련 질문 및 답변

1. predict함수 type 질문

   • 해당 질문에서는 predict 함수의 type 옵션 사용에 대해 설명하고 있습니다. 회귀 문제는 type을 지정하지 않으며, 분류 문제에서 확률 값을 요구할 경우에만 type='prob' 설정이 필요합니다.
     

2. 작업형제2유형 질문입니다.

   • 분류 문제에서 제출 파일의 형식이 어떻게 되어야 하는지에 대한 설명이 있습니다. 일반적으로 0과 1로 제출할 시 확률 대신 클래스를 반환하도록 설정해야 합니다.
     

기존 질문을 통해 자세한 부분 확인하시고, 혹시나 이어서 추가로 궁금하신 사항이 있으면 언제든 질문해 주세요.

저는 rjstkddl님의 이해를 돕고자 지식공유자님의 기존 답변을 참고하여 유사한 답변 링크를 추천해드리고 있습니다.
현재 베타 기능으로 답변이 만족스럽지 않을 수 있는 점 양해 부탁드립니다. 🙏
추가적으로 궁금한 점이 있으시면, 이어서 질문해 주세요. 곧 지식공유자께서 답변해 주실 것입니다.