example/reinforcement-learning/dqn/game.py

DEFAULT_MAX_EPISODE_STEP = 1000000

class Game(object):
    def __init__(self):
        self.total_reward = 0
        self.episode_reward = 0
        self.episode_step = 0
        self.max_episode_step = DEFAULT_MAX_EPISODE_STEP

    def start(self):
        raise NotImplementedError("Must Implement!")

    def begin_episode(self, max_episode_step):
        raise NotImplementedError("Must Implement!")

    @property
    def episode_terminate(self):
        raise NotImplementedError

    def get_observation(self):
        raise NotImplementedError

    @property
    def state_enabled(self):
        raise NotImplementedError

    def current_state(self):
        return self.replay_memory.latest_slice()

    def play(self, a):
        raise NotImplementedError