해결된 질문
작성
·
236
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] 로 할지는 어떤 것을 보고 정할 수 있는지 여쭈어볼 수 있을까요?