ML-based demand planning