Пример #1
0
 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)
Пример #2
0
 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)
Пример #3
0
 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)