회귀용 결정트리

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

Elie Kawerk

Data Scientist

Auto-mpg 데이터셋

auto

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

단일 특성의 Auto-mpg

auto1D

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

scikit-learn의 회귀트리

# Import DecisionTreeRegressor
from sklearn.tree import DecisionTreeRegressor
# Import train_test_split 
from sklearn.model_selection import train_test_split
# Import mean_squared_error as MSE
from sklearn.metrics import mean_squared_error as MSE

# Split data into 80% train and 20% test
X_train, X_test, y_train, y_test= train_test_split(X, y, 
                                                   test_size=0.2, 
                                                   random_state=3)

# Instantiate a DecisionTreeRegressor 'dt'
dt = DecisionTreeRegressor(max_depth=4, 
                           min_samples_leaf=0.1,
                           random_state=3)
Python으로 배우는 트리 기반 Machine Learning

scikit-learn의 회귀트리

# Fit 'dt' to the training-set
dt.fit(X_train, y_train)
# Predict test-set labels
y_pred = dt.predict(X_test)

# Compute test-set MSE
mse_dt =  MSE(y_test, y_pred)

# Compute test-set RMSE 
rmse_dt = mse_dt**(1/2)

# Print rmse_dt
print(rmse_dt) 

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

회귀트리의 정보 기준

불순도-회귀

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

예측

예측-회귀

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

선형회귀 vs. 회귀트리

auto1D

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

연습해 봅시다!

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

Preparing Video For Download...