def plot_data(logdir, i): recorder = Recorder() recorder.load(logdir) recorder.plot_ith_episode_curves(i)
def plot_and_save_ith_episode_data(logdir, i): recorder = Recorder() recorder.load(logdir) save_dir = logdir + '/episode{}/figs'.format(i) os.makedirs(save_dir, exist_ok=True) recorder.plot_and_save_ith_episode_curves(i, save_dir, True)
def plot_data(epi_num, logdir): recorder = Recorder() recorder.load(logdir) recorder.plot_mpc_rl(epi_num)