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
Show whitespace changes
Inline Side-by-side
convlab/policy/emoTUS/evaluate.py
+ 2
− 1
View file @ 5bedcba1
...@@ -241,7 +241,8 @@ def emotion_score(golden_emotions, gen_emotions): ...@@ -241,7 +241,8 @@ def emotion_score(golden_emotions, gen_emotions):
confusion_matrix=cm, display_labels=labels) confusion_matrix=cm, display_labels=labels)
disp.plot() disp.plot()
plt.savefig("emotion.png") 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) print(r)
return r return r
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment