augmented_prediction