INNER CODE UNIT · Python

on_move

rlcode/reinforcement-learning · 1-grid-world/2-value_iteration.py:81

    def on_move():
        display.move_along_policy(value_iteration.get_action)

    def on_clear():
        value_iteration.__init__(env)
        display.clear()
        display.agent_pos = [0, 0]
        display.clicks.clear()

    # Workflow: Calculate x several (until V stops changing) -> Print Policy -> Move.
    # Print Policy unlocks after Calculate; Move unlocks after Print Policy.
    display.buttons = [
        ("Calculate",    on_calculate),
        ("Print Policy", on_print_policy, lambda: display.click_count("Calculate") > 0),
        ("Move",         on_move,         lambda: display.click_count("Print Policy") > 0),
        ("Clear",        on_clear),
    ]
    display.mainloop()

View source record →