強化学習というのは、機械学習の一つの分野ですが、学習がうまくいっているのかどうか調べるのは難しい課題なんですよね。
そこで、Open AIのGymというツールがあります。
なんか、ちょいちょい見る、トンカチみたいなのがゆらゆら揺れてる動画…。これだったんだｗ
やり方は、次のサイトにある通りですが
英語を読むのも「ええい！面倒！！」ということもあると思いますので、超簡単に記載しておきます。
pip install gymでインスコします。
import gym
env = gym.make('CartPole-v0')
env.reset()
for _ in range(1000):
env.render()
env.step(env.action_space.sample()) # take a random action
これで走らせてみると、トンカチが一回転して、画面の外に消えちゃいますねｗ
他に、山を登る車、
'MountainCar-v0'
ってのもあります。さっきのgym.makeの中身を変えるだけです。
import gym
env = gym.make('MountainCar-v0')
env.reset()
for _ in range(1000):
env.render()
env.step(env.action_space.sample()) # take a random action
これも実行してみると、右往左往してまったく山を登りそうにない車が表示されます。
やはり、ランダムに何かを行ってみても何も結果は得られない… という人生の教訓を得た気になりますね！！
ちょっと進めて下記のようにします。
import gym
env = gym.make('CartPole-v0')
for i_episode in range(20):
observation = env.reset()
for t in range(100):
env.render()
print(observation)
action = env.action_space.sample()
observation, reward, done, info = env.step(action)
if done:
print("Episode finished after {} timesteps".format(t+1))
break
実行すると、ポールが倒れたり、画面の外に行ったりしません。
一瞬おお！と思いますが、落ち着こう。
observationというのをprintするようにしたので、次のようにprintされますが
[-0.06550298 -0.62622142 0.12089567 0.99951833]
[-0.07802741 -0.82273366 0.14088604 1.32759199]
[-0.09448208 -0.62964272 0.16743788 1.08210892]
[-0.10707494 -0.4370784 0.18908006 0.846299 ]
[-0.1158165 -0.24496922 0.20600604 0.6185299 ]
….
Episode finished after 16 timesteps
そもそも、この4つの数字なんだ！って話ですよね。
この答え、CartPoleのソースコード見ると書いてありました。（他のところにもかいてほしぃ…）
https://github.com/openai/gym/blob/master/gym/envs/classic_control/cartpole.py
Observation:
Type: Box(4)
Num Observation Min Max
0 Cart Position -4.8 4.8
1 Cart Velocity -Inf Inf
2 Pole Angle -24° 24°
3 Pole Velocity At Tip -Inf Inf
そして、そもそも終了条件ってなんなん？？
って思うと、これもソースコード見ると書いてありました。
Episode Termination:
Pole Angle is more than ±12°
Cart Position is more than ±2.4 (center of the cart reaches the edge of the display)
Episode length is greater than 200
Solved Requirements
Considered solved when the average reward is greater than or equal to 195.0 over 100 consecutive trials.
1.ポールの角度（さっきのObservationの3つ目ですね）が±12を超える
2.カートのポジション（さっきのObservationの１つ目ですね）が±2.4を超える
3.エピソードの長さが200以上
4.解決されたと思われる 平均的な報酬が100回の連続したトライアルの中で195を超える
報酬は、1回進めて、倒れなかったら1もらえるので、200回エピソードを繰り返して、195回目まで倒れなければ、そのモデルが優秀ということで、もうOKってことなんですかね。
Actionは左に行くか、右に行くかだけですが、これにVelocityなどが加わって、どのぐらいのポジションを進むかが決定されるようです。
倒れなくなったように見えますが、そうではなく、倒れようとしたところで（doneがTrueになる）env.resetされるので、倒れなくなったように見えるだけです。