From 0a706720428c1a75ff4a1e70e9062619bc61d8f1 Mon Sep 17 00:00:00 2001 From: linh <linh@hhu.de> Date: Wed, 25 Jan 2023 16:00:44 +0100 Subject: [PATCH] wip --- convlab/policy/emoTUS/self_bleu.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/convlab/policy/emoTUS/self_bleu.py b/convlab/policy/emoTUS/self_bleu.py index b77c4e11..37de963f 100644 --- a/convlab/policy/emoTUS/self_bleu.py +++ b/convlab/policy/emoTUS/self_bleu.py @@ -43,8 +43,7 @@ def calculate(candidates, bleu_mode="torch"): bleu = fast_bleu.SelfBLEU(sentences) x = bleu.get_score() # x = bleu.get_score() - print(x) - + print(sum(x[4])/len(x[4])) if __name__ == "__main__": args = arg_parser() -- GitLab