Entrenar el clasificador AdaBoost
Ahora que has instanciado el clasificador AdaBoost ada
, es hora de entrenarlo. También predecirás las probabilidades de obtener la clase positiva en el conjunto de pruebas. Esto puede hacerse de la siguiente manera:
Una vez entrenado el clasificador ada
, llama al método .predict_proba()
pasando X_test
como parámetro y extrae estas probabilidades troceando todos los valores de la segunda columna como sigue:
ada.predict_proba(X_test)[:,1]
El conjunto de datos Hígado indio se procesa para ti y 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 de etiquetas y_train
y y_test
están disponibles en tu espacio de trabajo. Además, también hemos cargado el
modelo instanciado ada
del ejercicio anterior.
Este ejercicio forma parte del curso
Machine learning con modelos basados en árboles en Python
Instrucciones de ejercicio
Ajusta
ada
al conjunto de entrenamiento.Evalúa las probabilidades de obtener la clase positiva en el conjunto de pruebas.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
# Fit ada to the training set
____
# Compute the probabilities of obtaining the positive class
y_pred_proba = ____.____(____)[____]