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
Anleitung zur Übung
- Importiere
DecisionTreeRegressor
vonsklearn.tree
. - Erstelle eine
DecisionTreeRegressor
dt
mit maximaler Tiefe 8 undmin_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
____.____(____, ____)