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