rldurham.plot_frozenlake

rldurham.plot_frozenlake(env, v=None, policy=None, trajectory=None, col_ramp=1, draw_vals=False, clear=False)

A helper function to draw the frozen lake environment with state values and policy.

Parameters:

• env – the specific instance of the environment

• v – state values

• policy – policy

• trajectory – a trajectory of states to plot

• col_ramp – non-linear rescaling for state values (values are mapped to grey scale using 1 - v ** col_ramp)

• draw_vals – show textual state values

• clear – clear output in IPython notebooks before showing plot