모델 인스턴스화하기
다음 연습 문제 세트에서는 회귀 트리의 편향과 분산 문제를 진단해 보겠습니다. 이번 연습에서 정의할 회귀 트리는 auto 데이터셋의 사용 가능한 모든 특징을 사용해 자동차의 mpg 소비를 예측합니다.
데이터 전처리는 이미 완료되었고, 작업 공간에 특징 행렬 X와 배열 y가 로드되어 있습니다. 또한 sklearn.tree에서 DecisionTreeRegressor 클래스가 임포트되어 있습니다.
이 연습은 강의의 일부입니다
Python으로 배우는 트리 기반 Machine Learning
연습 안내
sklearn.model_selection에서train_test_split을 임포트하세요.- 데이터를 학습 70%, 테스트 30%로 분할하세요.
- 최대 깊이를 4로 하고
min_samples_leaf를 0.26으로 설정한DecisionTreeRegressor를 인스턴스화하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# Import train_test_split from sklearn.model_selection
____
# Set SEED for reproducibility
SEED = 1
# Split the data into 70% train and 30% test
X_train, X_test, y_train, y_test = ____(____, ____, test_size=____, random_state=SEED)
# Instantiate a DecisionTreeRegressor dt
dt = ____(____=____, ____=____, random_state=SEED)