Commit c3d23644 authored by Ard Kastrati's avatar Ard Kastrati
Browse files

Changed the name to just loss

parent 127a5b95
......@@ -68,12 +68,12 @@ def plot_loss_torch(loss, output_directory, model):
def cp_dir(source, target):
call(['cp', '-a', source, target])
def comparison_plot():
def comparison_plot_accuracy():
run_dir = './results/OHBM/'
print(run_dir)
plt.figure()
plt.title('Comparison of the Validation accuracy' )
plt.title('Comparison of the validation accuracy' )
plt.grid(True)
plt.xlabel('epochs')
plt.ylabel('accuracy (%)')
......@@ -108,7 +108,7 @@ def comparison_plot_loss():
plt.plot(acc, '-', label=name)
plt.legend()
plt.savefig(run_dir + '/comparison_accuracy_loss.png')
plt.savefig(run_dir + '/comparison_loss.png')
def select_best_model():
......
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