LoslegenKostenlos loslegen

Trainiere deinen ersten Regressionsbaum

In dieser Übung trainierst du einen Regressionsbaum, um den mpg-Wert (Meilen pro Gallone) von Autos im auto-mpg-Datensatz anhand aller sechs verfügbaren Merkmale vorherzusagen.

Der Datensatz ist bereits aufbereitet und in 80 % Training und 20 % Test aufgeteilt. Die Merkmalsmatrix X_train und das Array y_train stehen dir in deinem Workspace zur Verfügung.

Diese Übung ist Teil des Kurses

Maschinelles Lernen mit baumbasierten Modellen in Python

Kurs anzeigen

Anleitung zur Übung

  • Importiere DecisionTreeRegressor aus sklearn.tree.
  • Erzeuge einen DecisionTreeRegressor dt mit einer maximalen Tiefe von 8 und min_samples_leaf auf 0.13 gesetzt.
  • Fitte dt auf den Trainingssatz.

Interaktive Übung

Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.

# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____

# Instantiate dt
____ = ____(max_depth=____,
             ____=____,
            random_state=3)

# Fit dt to the training set
____.____(____, ____)
Code bearbeiten und ausführen