LoslegenKostenlos loslegen

Trainiere deinen ersten Regressionsbaum

In dieser Übung trainierst du einen Regressionsbaum, um den mpg (Meilen pro Gallone)-Verbrauch der Autos im auto-mpg-Datensatz mit allen sechs verfügbaren Merkmalen vorherzusagen.

Der Datensatz wird für dich aufbereitet und in 80% Training und 20% Test aufgeteilt. Die Funktionen Matrix X_train und das Array y_train sind in deinem Arbeitsbereich verfügbar.

Diese Übung ist Teil des Kurses

Maschinelles Lernen mit baumbasierten Modellen in Python

Kurs anzeigen

Anleitung zur Übung

  • Importiere DecisionTreeRegressor von sklearn.tree.
  • Erstelle eine DecisionTreeRegressor dt mit maximaler Tiefe 8 und min_samples_leaf auf 0,13 gesetzt.
  • Passe dt an die Trainingsmenge an.

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