-
Notifications
You must be signed in to change notification settings - Fork 0
/
test.py
34 lines (29 loc) · 963 Bytes
/
test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
__author__ = 'tushar'
import gym
from agent import Agent
import json
with open('API.json') as api:
data = json.load(api)
env = gym.make('FrozenLake-v0')
env.monitor.start('/tmp/frozenlake-experiment-9')
observation = env.reset()
action_space = env.action_space
observation, reward, done, info = env.step(action_space.sample())
agent = Agent(observation, reward, info, action_space.sample(), action_space)
num_episodes = 5000
for i_episode in range(num_episodes):
observation = env.reset()
done = False
t = 0
while not done:
env.render()
action = agent.take_action()
(observation, reward, done, info) = env.step(action)
agent.update(observation, reward, info, action, action_space)
t = t + 1
if done:
break
print("Episode finished after {} timesteps".format(t+1))
env.monitor.close()
gym.scoreboard.api_key = data["api_key"]
gym.upload('/tmp/frozenlake-experiment-9')