시작하기무료로 시작하기

ROC 곡선

이제 랜덤 포레스트 분류기에 대한 ROC 곡선을 만들어 봅시다. 첫 단계는 .predict_proba() 메서드를 사용해 각 레이블에 대한 예측 확률을 계산하는 것입니다. 그런 다음 sklearn.metricsroc_curve 함수를 사용해 위양성률과 진양성률을 계산하고, 이를 matplotlib으로 시각화할 수 있어요.

훈련 세트 비율 70%로 학습된 RandomForestClassifier가 데이터에 이미 적합되어 있으며, 작업 공간에서 clf로 제공됩니다.

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

마케팅 애널리틱스: Python으로 고객 이탈 예측하기

강의 보기

실습형 인터랙티브 연습

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

# Generate the probabilities
y_pred_prob = ____.____(____)[:, 1]
코드 편집 및 실행