def __init__(self, task, agent):
Experiment.__init__(self, task, agent)
agent.learner.explorer.experiment = self
# agent.learner.module.getValue()
self.screen = pygame.display.set_mode(((xsize+2)*MAGNIFY,(ysize+2)*MAGNIFY))
pygame.display.set_caption('Policy Visualizer')
self.clock = pygame.time.Clock()
self.screenBuffer = pygame.Surface(self.screen.get_size())
self.screenBuffer = self.screenBuffer.convert()
self.screenBuffer.fill((64, 64, 64)) # Dark Gray
self.bombImage = pygame.image.load("bomb_image.png")
self.bombImage = pygame.transform.scale(self.bombImage, (MAGNIFY - 2, MAGNIFY - 2))
self.isPaused = False
self.isCrashed = False
self.speed = 10
self.num = 0
self.robotXA = -1
self.robotYA = -1
self.bomb_counter = 0
self.count = 0
self.acc_reward = 0
self.collect_data = False
if collect_data_file != None:
self.collect_data = True
self.collect_episode_data_file = open(collect_data_file + "_episodelen.data", "w")
self.collect_reward_data_file = open(collect_data_file + "_avg_reward.data", "w")
def __init__(self, task, agent):
Experiment.__init__(self, task, agent)
agent.learner.explorer.experiment = self
self.isPaused = False
self.isCrashed = False
self.speed = 10
self.num = 0
self.XA = 50
self.switch_state = 1
self.count = 0
self.acc_reward = 0
self.collect_data = False
if collect_data_file != None:
self.collect_data = True
self.collect_episode_data_file = open(
collect_data_file + "_episodelen.data", "w")
self.collect_reward_data_file = open(
collect_data_file + "_avg_reward.data", "w")
