Trainer – train.py

You can start training by using train.py.

Training

Go to blobrl directory

cd blobrl

start training

python train.py

Parameters

–agent:

StringDefault : agent_randomName of agent listed [agent_random, dqn, double_dqn, categorical_dqn]

–env:

StringDefault : CartPole-v1Name of gym environment listed in gyms.openai.com

–max_episode

IntegerDefault : 100Number of episode to train

–render

BooleanDefault : FalseShow render on each step or not

Exemples

Start training with DQN on CartPole-v1 with 1000 episodes and show environment

python train.py --agent dqn --env CartPole-v1 --render 1 --max_episode 1000