fix column name

NeuroDesk · May 1, 2024 · 4fdabbf · 4fdabbf
1 parent fdb6ac8
commit 4fdabbf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plot_results.py b/plot_results.py
@@ -37,7 +37,7 @@ def find_pareto_front(x, y):
             raise FileNotFoundError(f'{model} does not exist')
 
         dsc_csv = pd.read_csv(join(dirname, model + '_results', 'metrics.csv'), sep=',', usecols=['dsc'])
-        runtime_csv = pd.read_csv(join(model + '_results', 'running_time.csv'), sep=',', usecols=['time'])
+        runtime_csv = pd.read_csv(join(model + '_results', 'running_time.csv'), sep=',', usecols=['Running time (mean)'])
         # print(dsc_csv.values.flatten(), runtime_csv)
         dsc_mean = np.mean(dsc_csv.values.flatten())
         dsc.append(dsc_mean)