인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

Inflearn Community Q&A

munjihey1127's profile image
munjihey1127

asked

[After-work activities] Big Data Analysis Engineer Practical (Work-type 1,2,3)

Mock Exam 1 for Task Type 2

작업형 모의문제 1

Resolved

Written on

·

168

·

Edited

0

마지막에 예측및 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]이 정답인건가요?

python머신러닝빅데이터pandas빅데이터분석기사

Answer 1

0

roadmap님의 프로필 이미지
roadmap
Instructor

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

그래서

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

네 맞습니다.

그리고 roc-auc를 묻는다면 양성 클래스(1)의 확률값을 예측해 주세요

[기출] 예시문제 작업형2 영상을 참고해주세요

munjihey1127's profile image
munjihey1127

asked

Ask a question