Entrena el regresor GB
Ahora entrenarás el regresor de potenciación del gradiente gb que instanciaste en el ejercicio anterior y predecirás las etiquetas del conjunto de prueba.
El conjunto de datos se divide en un 80 % de entrenamiento y un 20 % de prueba. Las matrices de características X_train y X_test, así como las matrices y_train y y_test están disponibles en tu espacio de trabajo. Además, también hemos cargado la instancia del modelo gb que definiste en el ejercicio anterior.
Este ejercicio forma parte del curso
Machine learning con modelos basados en árboles en Python
Instrucciones del ejercicio
- Ajusta
gbal conjunto de entrenamiento. - Predice las etiquetas del conjunto de pruebas y asigna el resultado a
y_pred.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Fit gb to the training set
____
# Predict test set labels
y_pred = ____