Latih pohon regresi pertama Anda
Dalam latihan ini, Anda akan melatih pohon regresi untuk memprediksi konsumsi mpg (mil per galon) mobil pada auto-mpg dataset menggunakan keenam fitur yang tersedia.
Himpunan data telah diproses untuk Anda dan dibagi menjadi 80% latih dan 20% uji. Matriks fitur X_train dan array y_train tersedia di workspace Anda.
Latihan ini adalah bagian dari kursus
Machine Learning dengan Model Berbasis Pohon di Python
Petunjuk latihan
- Impor
DecisionTreeRegressordarisklearn.tree. - Instansiasi
DecisionTreeRegressordtdengan kedalaman maksimum 8 danmin_samples_leafdisetel ke 0.13. - Latih (
fit)dtpada himpunan latih.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____
# Instantiate dt
____ = ____(max_depth=____,
____=____,
random_state=3)
# Fit dt to the training set
____.____(____, ____)