AI for predictive analytics