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

이석민님의 프로필 이미지

작성한 질문수

[퇴근후딴짓] 빅데이터 분석기사 실기 (작업형1,2,3)

2회 기출유형(작업형2)

2회 작업형 2번 pred 질문 입니다.

해결된 질문

작성

·

233

0

from sklearn.tree import DecisionTreeClassifier
model = DecisionTreeClassifier(random_state = 2023)
model.fit(X_tr, y_tr)
pred = model.predict_proba(X_val)

roc_auc_score(y_val, pred[:,1])

안녕하세요?

마지막 pred 에서 슬라이싱할때.. 조금 헷갈리는데요..!

 

pred[:,1]로 하는 이유가 예측(시간에 맞춰 도착하지 않을 확률) 때문인거 같은데..!

혹시 pred[:,0]을 할지 pred[:,1] 로 할지는 어떤 것을 보고 정할 수 있는지 여쭈어볼 수 있을까요?

 

 

답변 1

0

퇴근후딴짓님의 프로필 이미지
퇴근후딴짓
지식공유자

문제에 도착하지 않을 확률 (1)이라고 명시되어 있었어요!
그에 맞춰 1 확률값을 제출했습니다.