This is the code for the ENE5AI deliverable 4.2: Active AI Agent, which accompanies the deliverable document.

Dependencies: Python >= 3.9, Poetry

Contents:

• main.py: main script
• time_series.py: time series prediction model
• environment.py: reinforcement learning environment
• agent.py: reinforcement learning model (AI agent)
• helpers.py: some helper functions

In order to run the project, fill in directory paths in main.py and run that script.