def _game_model(self, game_mode, game_name, action_space, getDate,
cnn_mode):
if game_mode == "drqn_training":
return DRQNTrainer(game_name, INPUT_SHAPE, action_space, getDate,
cnn_mode)
elif game_mode == "drqn_testing":
return DRQNSolver(game_name, INPUT_SHAPE, action_space, getDate,
cnn_mode)
elif game_mode == "ddrqn_training":
return DDRQNTrainer(game_name, INPUT_SHAPE, action_space, getDate,
cnn_mode)
elif game_mode == "ddrqn_testing":
return DDRQNSolver(game_name, INPUT_SHAPE, action_space, getDate,
cnn_mode)
elif game_mode == "ddqn_training":
return DDQNTrainer(game_name, INPUT_SHAPE, action_space, getDate,
cnn_mode)
elif game_mode == "ddqn_testing":
return DDQNSolver(game_name, INPUT_SHAPE, action_space, getDate,
cnn_mode)
elif game_mode == "dqn_training":
return DQNTrainer(game_name, INPUT_SHAPE, action_space, getDate,
cnn_mode)
elif game_mode == "dqn_testing":
return DQNSolver(game_name, INPUT_SHAPE, action_space, getDate,
cnn_mode)
elif game_mode == "ge_training":
return GETrainer(game_name, INPUT_SHAPE, action_space, getDate,
cnn_mode)
elif game_mode == "ge_testing":
return GESolver(game_name, INPUT_SHAPE, action_space, getDate,
cnn_mode)
else:
print("Unrecognized mode. Use --help")
exit(1)
def _game_model(self, game_mode, game_name, actions):
if game_mode == "ddqn_training":
return DDQNTrainer(game_name, INPUT_SHAPE, actions)
elif game_mode == "ddqn_testing":
return DDQNSolver(game_name, INPUT_SHAPE, actions)
elif game_mode == "ge_training":
return GETrainer(game_name, INPUT_SHAPE, actions)
elif game_mode == "ge_testing":
return GESolver(game_name, INPUT_SHAPE, actions)
else:
print "Unrecognized mode. Use --help"
exit(1)
def _game_model(self, action_space, INPUT_SHAPE=(4, 84, 84)):
if self.game_mode == "ddqn_training":
return DDQNTrainer(self.game_name, INPUT_SHAPE, action_space)
elif self.game_mode == "ddqn_testing":
return DDQNSolver(self.game_name, INPUT_SHAPE, action_space)
elif self.game_mode == "ge_training":
return GETrainer(self.game_name, INPUT_SHAPE, action_space)
elif self.game_mode == "ge_testing":
return GESolver(self.game_name, INPUT_SHAPE, action_space)
else:
print("Unrecognized mode. Use --help")
exit(1)