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

마지막에 예측및 csv제출 하실ㄸ ㅐ

'Attirition_Flag' : pred[:,1]로 하셨는데

이탈할 고객을 찾아라니깐 이탈의 확률인 1의 확률로 구해야하는거 아닌가요? 제가생각한 답은 'Attirition_Flag' : pred[:,0] 이였는데 왜 그런거죠?

+강사님

train['Attrition_Flag'].value_counts() 을 치면

0 6815

1 1286

이렇게 나오는데

pred  = model.predict_proba(X_val)

pred

array([[0.88, 0.12], [0.95, 0.05], [1. , 0. ], ..., [0.66, 0.34], [1. , 0. ], [0.93, 0.07]])

위 0,1 의 순서가 여기서도 동일하게 적용되서

여기서 0.88은 0의 확률 0.12는 1의 확률인가요?

그래서

'Attirition_Flag' : pred[:,1]이 정답인건가요?