diff --git a/convlab/policy/emoTUS/self_bleu.py b/convlab/policy/emoTUS/self_bleu.py index b77c4e115b953a12e24d55891c13f27974cb2c00..37de963ffd2addaf011cb826875a3a2b70889f89 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()