Machine learning based Power management