Träna SGB-regressorn
I den här övningen tränar du SGBR-modellen sgbr som du skapade i föregående övning och förutsäger etiketterna för testmängden.
Datamängden för cykeldelningsefterfrågan är redan inläst och förbehandlad; den är uppdelad i 80% träning och 20% test. Featuresmatriserna X_train och X_test, etikettvektorerna y_train och y_test, samt modellinstansen sgbr som du definierade i föregående övning finns tillgängliga i din arbetsyta.
Den här övningen är en del av kursen
Maskininlärning med trädbaserade modeller i Python
Övningsinstruktioner
- Anpassa
sgbrtill träningsdatan. - Förutsäg etiketterna för testmängden och tilldela resultaten till
y_pred.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Fit sgbr to the training set
____
# Predict test set labels
y_pred = ____