Predecir el abandono de estudiantes y mostrar cuándo la predicción sirve de algo
25.059 estudiantes internacionales, con datos que llegan en tres etapas: primero la solicitud, después la asistencia y por último los resultados de los módulos. Comparé XGBoost con una red neuronal en cada etapa, con una partición agrupada por estudiante y un conjunto de test reservado de 5.013.
La puntuación de 0,993 en la última etapa era demasiado buena: los expedientes imputados delataban la respuesta. Evaluado de nuevo sobre los 4.339 estudiantes que seguían en el curso, el modelo detecta 105 de 136 abandonos futuros. Una partición temporal mantuvo el ROC-AUC en 0,891, 0,904 y 0,980, pero mostró que el umbral de decisión tendría que recalibrarse con cada nueva promoción.
Python · XGBoost · Keras · scikit-learn · pandas