diff --git a/baselines/results_plotter.py b/baselines/results_plotter.py index 34ac898..057f946 100644 --- a/baselines/results_plotter.py +++ b/baselines/results_plotter.py @@ -72,9 +72,9 @@ def plot_results(dirs, num_timesteps, xaxis, yaxis, task_name): plot_curves(xy_list, xaxis, yaxis, task_name) # Example usage in jupyter-notebook -# from baselines import log_viewer +# from baselines import results_plotter # %matplotlib inline -# log_viewer.plot_results(["./log"], 10e6, log_viewer.X_TIMESTEPS, "Breakout") +# results_plotter.plot_results(["./log"], 10e6, results_plotter.X_TIMESTEPS, "Breakout") # Here ./log is a directory containing the monitor.csv files def main():