GitHub - Curtin-Machine-Learning-Club/CartPoleAI

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
.idea		.idea
.ipynb_checkpoints		.ipynb_checkpoints
__pycache__		__pycache__
model		model
Agent.py		Agent.py
CartPole.ipynb		CartPole.ipynb
DeepQNet.py		DeepQNet.py
QTrainer.py		QTrainer.py
README.txt		README.txt
ReplayMemory.py		ReplayMemory.py

Repository files navigation

CartPole AI
 - Author : Milan Marocchi
 - References : https://gym.openai.com/envs/CartPole-v0/
                https://pytorch.org/tutorials/intermediate/reinforcement_q_learning.html
                https://github.com/Lazydok/RL-Pytorch-cartpole
 - Notes : The code is based off of the references.

How to run:
    python3 Agent.py run

How to train:
    python3 Agent.py train <steps>