Example #1
0
class BaseModel:
    def __init__(self, game_name, mode_name, log_directory, input_shape,
                 action_space):
        self.logger = Logger('%s_%s' % (game_name, mode_name), log_directory)
        self.input_shape = input_shape
        self.action_space = action_space

    def step_update(self, total_step):
        pass

    def save_run(self, score, step):
        self.logger.add_score(score)
        self.logger.add_step(step)

    def move(self, state):
        pass
Example #2
0
class BaseGameModel:
    def __init__(self, game_name, mode_name, logger_path, input_shape,
                 action_space):
        self.action_space = action_space
        self.input_shape = input_shape
        self.logger = Logger(game_name + " " + mode_name, logger_path)

    def save_run(self, score, step, run):
        self.logger.add_score(score)
        self.logger.add_step(step)
        self.logger.add_run(run)

    def get_move(self, state):
        pass

    def move(self, state):
        pass

    def remember(self, state, action, reward, next_state, done):
        pass

    def step_update(self, total_step):
        pass