1. 학습
    2. /
  2. 강의
    3. /
  3. Python으로 배우는 트리 기반 Machine Learning
Connected

연습 문제

엔트로피 vs 지니 지수

이 연습 문제에서는 dt_entropy의 테스트 세트 정확도를 또 다른 트리 dt_gini의 정확도와 비교해 보겠습니다. 트리 dt_gini는 동일한 데이터셋과 동일한 하이퍼파라미터로 학습했으며, 정보 기준만 키워드 'gini'를 사용해 지니 지수로 설정했습니다.

X_test, y_test, dt_entropy, 그리고 dt_gini가 달성한 테스트 세트 정확도에 해당하는 accuracy_gini가 작업 공간에 준비되어 있습니다.

지침

100 XP
  • sklearn.metrics에서 accuracy_score를 임포트하세요.
  • dt_entropy의 테스트 세트 레이블을 예측하고 결과를 y_pred에 할당하세요.
  • dt_entropy의 테스트 세트 정확도를 평가하고 결과를 accuracy_entropy에 할당하세요.
  • accuracy_entropy와 accuracy_gini를 확인하세요.