시작하기무료로 시작하기

최적 트리 평가하기

이번 연습 문제에서는 grid_dt의 최적 모델에 대해 테스트 세트 ROC AUC 점수를 평가해 보겠습니다.

이를 위해 먼저 테스트 세트 각 관측치에 대해 양성 레이블이 될 확률을 구해야 합니다. sklearn 분류기의 predict_proba() 메서드를 사용하면 음성/양성 클래스 레이블의 확률이 각각 열에 담긴 2차원 배열을 계산할 수 있어요.

데이터셋은 이미 로드 및 전처리(수치형 특성 표준화)되어 있으며, 80% 학습/20% 테스트로 분할되어 있습니다. 작업 공간에는 X_test, y_test가 준비되어 있습니다. 또한 이전 연습 문제에서 생성한 학습된 GridSearchCV 객체 grid_dt도 로드되어 있어요. 참고로 grid_dt는 다음과 같이 학습했습니다:

grid_dt.fit(X_train, y_train)

이 연습은 강의의 일부입니다

Python으로 배우는 트리 기반 Machine Learning

강의 보기

연습 안내

  • sklearn.metrics에서 roc_auc_score를 임포트하세요.

  • grid_dt에서 .best_estimator_ 속성을 추출해 best_model에 할당하세요.

  • 테스트 세트에 대해 양성 클래스의 예측 확률을 y_pred_proba로 구하세요.

  • best_model의 테스트 세트 ROC AUC 점수 test_roc_auc를 계산하세요.

실습형 인터랙티브 연습

이 예제를 이 샘플 코드를 완성하여 풀어보세요.

# Import roc_auc_score from sklearn.metrics
____

# Extract the best estimator
best_model = ____

# Predict the test set probabilities of the positive class
y_pred_proba = ____

# Compute test_roc_auc
test_roc_auc = ____

# Print test_roc_auc
print('Test set ROC AUC score: {:.3f}'.format(test_roc_auc))
코드 편집 및 실행