Trainiere den GB Regressor
Jetzt trainierst du den Gradient-Boosting-Regressor gb
, den du in der vorherigen Übung instanziiert hast, und sagst die Beschriftungen der Testmenge voraus.
Der Datensatz wird in 80% Training und 20% Test aufgeteilt. Die Merkmalsmatrizen X_train
und X_test
, sowie die Arrays y_train
und y_test
sind in deinem Arbeitsbereich verfügbar. Außerdem haben wir auch die Modellinstanz gb
geladen, die du in der vorherigen Übung definiert hast.
Diese Übung ist Teil des Kurses
Maschinelles Lernen mit baumbasierten Modellen in Python
Anleitung zur Übung
- Passe
gb
an die Trainingsmenge an. - Sage die Beschriftungen der Testmenge voraus und ordne das Ergebnis
y_pred
zu.
Interaktive Übung zum Anfassen
Probieren Sie diese Übung aus, indem Sie diesen Beispielcode ausführen.
# Fit gb to the training set
____
# Predict test set labels
y_pred = ____