AI to predict electricity production and consumption