Using a small dense NN. Tested gyms with good results are:
- CartPole-v1
- FlappyBird-v0
- Snake-v0
Using virtualenv is recomended
python -m venv venv
./venv/bin/activate
Install dependancies
pip install --force-reinstall -r requirements.txt
To train the model:
python ./gym_train.py
To run the model:
python ./gym_play.py