Skip to content
Snippets Groups Projects
Commit 5bedcba1 authored by Hsien-Chin Lin's avatar Hsien-Chin Lin
Browse files

wip

parent 0fad267f
No related branches found
No related tags found
No related merge requests found
......@@ -241,7 +241,8 @@ def emotion_score(golden_emotions, gen_emotions):
confusion_matrix=cm, display_labels=labels)
disp.plot()
plt.savefig("emotion.png")
r = {"macro_f1": macro_f1, "sep_f1": list(sep_f1), "cm": list(cm)}
r = {"macro_f1": macro_f1, "sep_f1": list(
sep_f1), "cm": [list(c) for c in list(cm)]}
print(r)
return r
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment