machine learning-based algorithm for prediction