7.3.3 회귀 트리 알고리즘
회귀 트리 알고리즘은 타깃 변수가 카테고리형이 아닌 연속형 변수라는 점만 제외하면 분류 트리 알고리즘과 비슷합니다.
회귀 문제에 회귀 트리 알고리즘 적용하기
이 절에서는 회귀 트리 알고리즘으로 회귀 문제를 풀어 보겠습니다.
1. 회귀 트리 모델을 훈련합니다.
▼ myRegressionTree.ipynb
[in :]
from sklearn.tree import DecisionTreeRegressor
regressor = DecisionTreeRegressor(max_depth=3)
regressor.fit(X_train, y_train)
[out:]
DecisionTreeRegressor(criterion='mse', max_depth=3, max_features=None,
max_leaf_nodes=None, min_impurity_decrease=0.0,
min_impurity_split=None, min_samples_leaf=1,
min_samples_split=2, min_weight_fraction_leaf=0.0,
presort=False, random_state=None, splitter='best')