시작하기무료로 시작하기

기준으로 엔트로피 사용하기

이번 연습에서는 Wisconsin Breast Cancer 데이터셋에 대해 정보 기준으로 엔트로피를 사용해 분류 트리를 학습해 보겠습니다. 데이터셋의 30개 모든 특성을 사용하며, 데이터는 학습 80%, 테스트 20%로 분할되어 있습니다.

작업 공간에는 X_train과 레이블 배열 y_train이 제공되어 있습니다.

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

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

강의 보기

연습 안내

  • sklearn.tree에서 DecisionTreeClassifier를 임포트하세요.

  • 최대 깊이가 8인 DecisionTreeClassifierdt_entropy라는 이름으로 인스턴스화하세요.

  • 정보 기준을 'entropy'로 설정하세요.

  • 학습 세트에 dt_entropy를 학습(fit)하세요.

실습형 인터랙티브 연습

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

# Import DecisionTreeClassifier from sklearn.tree
from ____.____ import ____

# Instantiate dt_entropy, set 'entropy' as the information criterion
dt_entropy = ____(____=____, ____='____', random_state=1)

# Fit dt_entropy to the training set
____.____(____, ____)
코드 편집 및 실행