Träna GB-regressorn
Nu ska du träna gradient boosting-regressorn gb som du skapade i föregående övning och prediktera etiketter för testmängden.
Datamängden är uppdelad i 80% träning och 20% test. Särdragsmatriserna X_train och X_test, samt arrayerna y_train och y_test, finns tillgängliga i din miljö. Dessutom har modellinstansen gb som du definierade i föregående övning laddats in.
Den här övningen är en del av kursen
Maskininlärning med trädbaserade modeller i Python
Övningsinstruktioner
- Anpassa
gbtill träningsdata. - Prediktera etiketterna för testmängden och tilldela resultatet till
y_pred.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Fit gb to the training set
____
# Predict test set labels
y_pred = ____