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 merupakan bagian dari kursus
Machine Learning dengan Model Berbasis Pohon di Python
Instruksi latihan
- Impor
DecisionTreeRegressordarisklearn.tree. - Instansiasi
DecisionTreeRegressordtdengan kedalaman maksimum 8 danmin_samples_leafdisetel ke 0.13. - Latih (
fit)dtpada himpunan latih.
Latihan interaktif langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____
# Instantiate dt
____ = ____(max_depth=____,
____=____,
random_state=3)
# Fit dt to the training set
____.____(____, ____)