AI-driven forecasting