From d8c4eecb1875a24b0fecebfd1e13aa4a27756253 Mon Sep 17 00:00:00 2001 From: Niko Date: Thu, 1 Jun 2023 15:46:13 +0200 Subject: [PATCH] debugging --- plotter/reward_plotter.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/plotter/reward_plotter.py b/plotter/reward_plotter.py index 3b546ea..257ba08 100644 --- a/plotter/reward_plotter.py +++ b/plotter/reward_plotter.py @@ -12,8 +12,8 @@ def plot_csv(paths, x_axis, y_axis): x = np.linspace(0, mean.shape[0], mean.shape[0]) # Extract the first part of the filename and use it as a label - label = os.path.basename(path_).split('-')[0:3] - label = f"{label[0]} {float(label[1].replace('_','.'))}, nrbfs = {int(label[2])}" + label = os.path.basename(path_).split('-')[0:5] + label = f"{label[1]}, {label[2]}, {float(label[3].replace('_','.'))}, nrbfs = {int(label[4])}" plt.plot(x, mean, label=label) plt.fill_between( @@ -32,7 +32,9 @@ def plot_csv(paths, x_axis, y_axis): if __name__ == '__main__': - filenames = ['random-1_0-5-1685552722_2243946.csv'] + filenames = ['mc-ei-random-1_0-5-1685622201_6965265.csv', + 'mc-pi-random-1_0-5-1685622464_9843714.csv', + 'mc-cb-random-1_0-5-1685622728_8990934.csv'] home_dir = os.path.expanduser('~') file_path = os.path.join(home_dir, 'Documents/IntRLResults') paths = [os.path.join(file_path, filename) for filename in filenames]