MulaiMulai sekarang secara gratis

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

Lihat Kursus

Petunjuk latihan

  • Impor DecisionTreeRegressor dari sklearn.tree.
  • Instansiasi DecisionTreeRegressor dt dengan kedalaman maksimum 8 dan min_samples_leaf disetel ke 0.13.
  • Latih (fit) dt pada 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
____.____(____, ____)
Edit dan Jalankan Kode