Measuring ROC AUC in a custom callback