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")