Merge pull request #9 from nicolas-lair/patch-1

Allow verbose off for Bleu score
salaniz · Nov 18, 2020 · 14f4a1a · 14f4a1a
2 parents 38c2534 + 6862087
commit 14f4a1a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bleu/bleu.py b/bleu/bleu.py
@@ -18,7 +18,7 @@ def __init__(self, n=4):
         self._hypo_for_image = {}
         self.ref_for_image = {}
 
-    def compute_score(self, gts, res):
+    def compute_score(self, gts, res, verbose=1):
 
         assert(gts.keys() == res.keys())
         imgIds = gts.keys()
@@ -37,7 +37,7 @@ def compute_score(self, gts, res):
             bleu_scorer += (hypo[0], ref)
 
         #score, scores = bleu_scorer.compute_score(option='shortest')
-        score, scores = bleu_scorer.compute_score(option='closest', verbose=1)
+        score, scores = bleu_scorer.compute_score(option='closest', verbose=verbose)
         #score, scores = bleu_scorer.compute_score(option='average', verbose=1)
 
         # return (bleu, bleu_info)