extract_predictions