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
Anleitung zur Übung
- Importiere
DecisionTreeRegressoraussklearn.tree. - Erzeuge einen
DecisionTreeRegressordtmit einer maximalen Tiefe von 8 undmin_samples_leafauf 0.13 gesetzt. - Fitte
dtauf 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
____.____(____, ____)