Production forecast using AI and machine learning