1. 학습
  2. /
  3. 강의
  4. /
  5. Python으로 배우는 Explainable AI

Connected

연습 문제

결정 트리로 특성 중요도 계산하기

심장 질환 데이터셋을 사용해 심장 질환 위험 환자를 식별하는 결정 트리 분류기를 만들었어요. 이제 모델을 설명하기 위해 특성 중요도를 분석해 심장 질환 예측의 핵심 요인을 파악하고, 보다 정밀한 의료 개입이 가능하도록 해야 해요.

matplotlib.pyplot은 plt로 임포트되어 있어요. X_train과 y_train은 미리 로드되어 있어요.

지침

100 XP
  • model에서 특성 중요도를 추출하세요.
  • 주어진 feature_names에 대해 feature_importances를 그래프로 표시하세요.