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 de ejercicio
- Ajusta
gb
al conjunto de entrenamiento. - Predice las etiquetas del conjunto de pruebas y asigna el resultado a
y_pred
.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
# Fit gb to the training set
____
# Predict test set labels
y_pred = ____