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