ai-state-prediction