시작하기무료로 시작하기

SHAP으로 심장 질환의 핵심 예측 변수 찾기

여러분의 과제는 SHAP을 사용해, 미리 학습된 RandomForestClassifier model에서 각 특성(feature)이 심장 질환 예측에 어떤 영향을 미치는지 이해하는 것입니다.

특성을 담은 X, 레이블을 담은 y, 그리고 랜덤 포레스트 분류기 model은 미리 로드되어 있습니다.

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

Python으로 배우는 Explainable AI

강의 보기

연습 안내

  • explainer라는 이름의 SHAP tree explainer를 생성하세요.
  • shap_values를 계산하세요.
  • 평균 절대 SHAP 값 mean_abs_shap을 계산하세요.

실습형 인터랙티브 연습

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

import shap

# Create a SHAP Tree Explainer
explainer = ____

# Calculate SHAP values
shap_values = ____

# Calculate mean absolute SHAP values
mean_abs_shap = ____

plt.bar(X.columns, mean_abs_shap)
plt.title('Mean Absolute SHAP Values for RandomForest')
plt.xticks(rotation=45)
plt.show()
코드 편집 및 실행