def test_no_enemies_ppo2(self): print('testing test_no_enemies_ppo2') seed = 42 path = 'models/AmidarToyboxNoFrameskip-v4.regress.model' # # You need to do this if you want to load more than one model with tensorflow with tf.Session(graph=tf.Graph()): model = oai.getModel(self.env, 'ppo2', seed, path) self.runTest(model)
def test_scenario_ppo2(self): seed = 42 fdir = os.path.dirname(os.path.abspath(__file__)) # path = os.sep.join([fdir, 'models', 'AmidarToyboxNoFrameskip-v4.ppo2.5e7.3771075072.2019-05-18.model']) path = 'models/AmidarToyboxNoFrameskip-v4.regress.model' model = oai.getModel(self.env, 'ppo2', seed, path) # Set low to be a test of a test! self.runTest(model)
def test_ezchannel_ppo2(self): seed = 8675309 path = 'models/BreakoutToyboxNoFrameskip-v4.regress.model' with brk.BreakoutIntervention(self.getToybox()) as intervention: bricks = intervention.game.bricks with tf.Session(graph=tf.Graph()): model = oai.getModel(self.env, 'ppo2', seed, path) bricks = random.choice(bricks, 2) self.runTest(model, collection=bricks)
def test_no_enemies_all_models(self): seed = 42 fdir = os.path.dirname(os.path.abspath(__file__)) models = [ f for f in os.listdir(fdir + os.sep + 'models') if f.startswith('Amidar') ] print('num models:', len(models)) for trained in models: print(trained) path = os.sep.join([fdir, 'models', trained]) family = trained.split('.')[1] with tf.Session(graph=tf.Graph()): model = oai.getModel(self.env, family, seed, path) self.runTest(model)
def test_polar_starts_regress(self): with tf.Session(graph=tf.Graph()): model = oai.getModel(self.env, 'ppo2', seed, path) degs = random.choice(range(0, 360, 5), 2) self.runTest(model, collection=degs)
def test_shieldxs_ppo2(self): model = oai.getModel(self.env, 'ppo2', seed, path) shield_configs = [(1,0,0), (0,1,0), (0,0,1), (0,0,0), (1,1,1)] self.xs_observed = Counter() self.runTest(model, collection=shield_configs)
def test_no_shields_ppo2(self): model = oai.getModel(self.env, 'ppo2', seed, path) ps = random.choice([n/10 for n in range(11)], 2) self.runTest(model, collection=ps)
def test_no_shields_ppo2(self): model = oai.getModel(self.env, 'ppo2', seed, path) self.runTest(model)