To receive notifications about scheduled maintenance, please subscribe to the mailing-list gitlab-operations@sympa.ethz.ch. You can subscribe to the mailing-list at https://sympa.ethz.ch

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.grid(True)
plt.xlabel('epochs')
plt.ylabel('loss (%)')
plt.ylabel('loss')
for experiment in os.listdir(run_dir):
name = experiment
print(name)
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)
plt.legend()
......
Markdown is supported
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