Trainer – train.py¶
You can start training by using 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