Commit 20322684 authored by Colleen Beaumard's avatar Colleen Beaumard
Browse files

Modification of scoring.py

parent 69b6d69f
......@@ -227,8 +227,9 @@ ax.set_xlabel("Epochs")
ax.set_ylabel("CrossEntropyLoss()")
twin.set_ylabel("Validation accuracy (%)")
ax.set_xlim(0,len(aepoch))
twin.set_ylim(0, 100)
ax.set_ylim(0,9)
ax.set_xlim(0, len(aepoch))
if len(reduce_lr_list) != 0:
reduce_lr_list = [re.sub(r"\s+", " ", elmt) for elmt in reduce_lr_list]
......@@ -243,7 +244,7 @@ if len(reduce_lr_list) != 0:
i += 1
ax.legend(handles=handles)
#plt.gca().xaxis.set_major_locator(matplotlib.ticker.MaxNLocator(20))
plt.title("Model: {}{}{}_{}emo_{}batch\nEpoch: {} lr: {} Data: Test-IEMOCAP {}".format(model_type, model_2nd, freeze, cates, nb_batch, aepoch[-1], lr, ses_nb))
plt.savefig(os.path.join(path, "losses_{}{}{}_{}emo_{}batch_epoch-{}_lr-{}_Test-IEMOCAP{}.png".format(model_type, model_2nd, freeze, cates, nb_batch, aepoch[-1], lr, ses_nb)))
plt.show()
......
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