This project implements a reinforcement learning environment for a cleaning robot using OpenAI's Gym framework. The environment simulates a grid where a robot can move, clean dirt, and create dirt under certain conditions.
-
Clone the repository:
git clone https://github.com/AISC10-SSO/cleaning_robot.git cd cleaning_robot_env
-
Create a virtual environment:
python -m venv venv source venv/bin/activate # On Windows use `venv\Scripts\activate`
-
Install the required dependencies:
pip install -e .
To train the agent, run the train.py script:
python cleaning_robot_env/train.py