Pohon Keputusan untuk Regresi

Machine Learning dengan Model Berbasis Pohon di Python

Elie Kawerk

Data Scientist

Dataset Auto-mpg

auto

Machine Learning dengan Model Berbasis Pohon di Python

Auto-mpg dengan satu fitur

auto1D

Machine Learning dengan Model Berbasis Pohon di Python

Pohon Regresi di 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)
Machine Learning dengan Model Berbasis Pohon di Python

Pohon Regresi di 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
Machine Learning dengan Model Berbasis Pohon di Python

Kriteria Informasi untuk Pohon Regresi

impurity-reg

Machine Learning dengan Model Berbasis Pohon di Python

Prediksi

pred-reg

Machine Learning dengan Model Berbasis Pohon di Python

Regresi Linear vs. Pohon Regresi

auto1D

Machine Learning dengan Model Berbasis Pohon di Python

Ayo berlatih!

Machine Learning dengan Model Berbasis Pohon di Python

Preparing Video For Download...