Source code for simple_rl.planning.PlannerClass

[docs]class Planner(object): ''' Abstract class for a Planner. ''' def __init__(self, mdp, name="planner"): self.name = name # MDP components. self.mdp = mdp self.init_state = self.mdp.get_init_state() self.states = set([]) self.actions = mdp.get_actions() self.reward_func = mdp.get_reward_func() self.transition_func = mdp.get_transition_func() self.gamma = mdp.gamma self.has_planned = False def plan(self, state): pass def policy(self, state): pass def __str__(self): return self.name