/cartpole-tf-dqn

Play CartPole with DQN implemented with Tensorflow 2.0

Primary LanguageJupyter NotebookMIT LicenseMIT

CartPole with TF 2.0

This repository demonstrates how to implement a DQN reinforcement learning algorithm to play the CartPole game with TensorFlow 2.0.

Getting started

Run in Colab

Open colab_script.ipynb and then click on the badge Open in Colab located on the top of the notebook.

Run locally

First create a virtual environment (recommended).

Then, run the following command with the virtual environment is activated:

pip install -r requirements.txt

Start training

python entrypoint.py

Run in streamlit

streamlit run entrypoint.py