A framework for multi-stage ML-based electricity demand forecasting