Parameter setzen und ein Modell fitten
Vorhersageaufgaben fallen in zwei Kategorien: Regression oder Klassifikation. Im Candy-Datensatz ist das Ergebnis eine kontinuierliche Variable, die beschreibt, wie oft eine Süßigkeit in einer Reihe von 1-gegen-1-Duellen einer anderen vorgezogen wurde. Um diesen Wert (die Gewinnquote) vorherzusagen, verwendest du ein Regressionsmodell.
In dieser Übung legst du einige Parameter für ein Random-Forest-Regressionsmodell rfr fest.
Diese Übung ist Teil des Kurses
Modellvalidierung in Python
Anleitung zur Übung
- Füge
rfreinen Parameter hinzu, sodass die Anzahl der Bäume 100 beträgt und die maximale Tiefe dieser Bäume 6 ist. - Sorge für Reproduzierbarkeit, indem du den Zufallszustand (
random_state) auf1111setzt. - Verwende die Methode
.fit(), um das Random-Forest-Regressionsmodell mitX_trainals Eingabedaten undy_trainals Zielvariable zu trainieren.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# Set the number of trees
rfr.____ = ____
# Add a maximum depth
rfr.____ = ____
# Set the random state
rfr.____ = ____
# Fit the model
rfr.____(____, ____)