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

wip

parent eedb08cc
Branches
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
convlab/policy/emoTUS/evaluate.py
+ 3
− 3
View file @ 0d92d7ec
...@@ -219,9 +219,9 @@ class Evaluator: ...@@ -219,9 +219,9 @@ class Evaluator:
result[metric] = sum(scores[metric])/len(scores[metric]) result[metric] = sum(scores[metric])/len(scores[metric])
print(f"{metric}: {result[metric]}") print(f"{metric}: {result[metric]}")
emo_score = emotion_score(golden_emotions, gen_emotions) emo_score = emotion_score(golden_emotions, gen_emotions)
for metric in emo_score: # for metric in emo_score:
result[metric] = emo_score[metric] # result[metric] = emo_score[metric]
print(f"{metric}: {result[metric]}") # print(f"{metric}: {result[metric]}")
result["dialog"] = dialog_result result["dialog"] = dialog_result
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment