INNER CODE UNIT · Python
get_value
rlcode/reinforcement-learning · 1-grid-world/1-policy_iteration.py:95
def get_value(self, state):
return round(self.value_table[state[0]][state[1]], 2)
if __name__ == "__main__":
env = Env()
policy_iteration = PolicyIteration(env)
display = GraphicDisplay(policy_iteration, title="Policy Iteration")
def on_evaluate():
policy_iteration.policy_evaluation()
display.show_values(policy_iteration.value_table)
def on_improve():
policy_iteration.policy_improvement()
display.show_arrows(policy_iteration.policy_table)
def on_move():
display.move_along_policy(policy_iteration.get_action)