mirror of
https://github.com/wassname/DeepRL.git
synced 2026-09-09 11:13:47 +08:00
Fix import
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
# Adapted from https://github.com/openai/baselines/blob/master/baselines/results_plotter.py
|
||||
|
||||
from ..component.bench import load_monitor_log
|
||||
import numpy as np
|
||||
from ..component import *
|
||||
import os
|
||||
import re
|
||||
|
||||
@@ -44,7 +44,7 @@ class Plotter:
|
||||
def load_results(self, dirs, max_timesteps=1e8, x_axis=X_TIMESTEPS, episode_window=100):
|
||||
tslist = []
|
||||
for dir in dirs:
|
||||
ts = component.load_monitor_log(dir)
|
||||
ts = load_monitor_log(dir)
|
||||
ts = ts[ts.l.cumsum() <= max_timesteps]
|
||||
tslist.append(ts)
|
||||
xy_list = [self.ts2xy(ts, x_axis) for ts in tslist]
|
||||
|
||||
Reference in New Issue
Block a user