(OpenAI) DQN

OpenAIのBaselinesでDQNを試してみた。(ほぼReadme)

導入

baselineを入れろとのことだがtensorflowがgpu版ではないので、後で入れ替える。 あとopencvとgym[atari]も必要。

pip install baseline
pip uninstall tensorflow
pip install tensorflow-gpu
conda install --channel https://conda.anaconda.org/menpo opencv3
pip install gym[atari]

テスト

cartpole

python -m baselines.deepq.experiments.train_cartpole

で学習、

python -m baselines.deepq.experiments.enjoy_cartpole

でテスト。

pong

python -m baselines.deepq.experiments.train_pong

で学習、

python -m baselines.deepq.experiments.enjoy_pong

でテスト。