Commit 494c3854 authored by Ard Kastrati's avatar Ard Kastrati
Browse files

Changed the comparison plot for loss without percentage

parent 06ded0d8
......@@ -2,7 +2,7 @@ from config import config
from ensemble import run
import numpy as np
import scipy
from utils.utils import select_best_model, comparison_plot, comparison_plot_loss
from utils.utils import select_best_model, comparison_plot_accuracy, comparison_plot_loss
from utils import IOHelper
from scipy import io
import h5py
......@@ -97,14 +97,14 @@ def comparison_plot_loss():
plt.title('Comparison of the validation loss')
plt.ylabel('loss (%)')
for experiment in os.listdir(run_dir):
name = experiment
if (name != 'eegnet'):
summary = pd.read_csv(run_dir + experiment + '/' + name + '_history.csv')
acc = 100 * summary['val_loss']
acc = summary['val_loss']
plt.plot(acc, '-', label=name)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment