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

wip

parent e56d1630
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
+ 1
− 1
View file @ 58da110b
...@@ -235,7 +235,7 @@ def emotion_score(golden_emotions, gen_emotions): ...@@ -235,7 +235,7 @@ def emotion_score(golden_emotions, gen_emotions):
macro_f1 = metrics.f1_score(golden_emotions, gen_emotions, average="macro") macro_f1 = metrics.f1_score(golden_emotions, gen_emotions, average="macro")
sep_f1 = metrics.f1_score( sep_f1 = metrics.f1_score(
golden_emotions, gen_emotions, average=None, labels=labels) golden_emotions, gen_emotions, average=None, labels=labels)
cm = metrics.confusion_matrix(golden_emotions, gen_emotions, labels) cm = metrics.confusion_matrix(golden_emotions, gen_emotions, labels=labels)
disp = metrics.ConfusionMatrixDisplay( disp = metrics.ConfusionMatrixDisplay(
confusion_matrix=cm, display_labels=labels) confusion_matrix=cm, display_labels=labels)
disp.plot() disp.plot()
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment