Update bleu.py

Allow a parameter in compute score method to turn off verbose in Bleu_scorer compute_score method
salaniz · Nov 18, 2020 · 6862087 · 6862087
1 parent 38c2534
commit 6862087
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)