/
Blocks_World_Random_Agent.py
49 lines (39 loc) · 1.04 KB
/
Blocks_World_Random_Agent.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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
# -*- coding: utf-8 -*-
"""
Spyder Editor
This is a temporary script file.
"""
import sys
#sys.path.append("/Users/rubengarzon/Documents/Projects/phD/Repo/gym")
import random
import gym
numBlocks = 3
env = gym.make('BlocksWorld-v0')
env.seed(0)
env.reset()
done = False
num_episodes = 1000
ep_lengths = []
n = 0
while (n<num_episodes):
steps =1
done = False
env.reset()
while (done == False):
next_action = [random.randint(0,numBlocks),random.randint(0,numBlocks)]
obs, reward, done, empty = env.step (next_action)
print ('Next action ' + str(next_action))
print ('Obs ' + str(obs))
#env.render()
steps +=1
print (done)
print ('New episode')
ep_lengths.append(steps)
n+=1
print ("Average episode length " + str(sum(ep_lengths) / float(len(ep_lengths))))
#input("Press Enter to continue...")
#Average episode length with nBlocks=2 19,37
#Average episode length with nBlocks=3 212,359
#Average episode length with nBlocks=4 1673,978
#env.reset()
#env.render()