Skip to content

Commit 14f4a1a

Browse files
authored
Merge pull request #9 from nicolas-lair/patch-1
Allow verbose off for Bleu score
2 parents 38c2534 + 6862087 commit 14f4a1a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bleu/bleu.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def __init__(self, n=4):
1818
self._hypo_for_image = {}
1919
self.ref_for_image = {}
2020

21-
def compute_score(self, gts, res):
21+
def compute_score(self, gts, res, verbose=1):
2222

2323
assert(gts.keys() == res.keys())
2424
imgIds = gts.keys()
@@ -37,7 +37,7 @@ def compute_score(self, gts, res):
3737
bleu_scorer += (hypo[0], ref)
3838

3939
#score, scores = bleu_scorer.compute_score(option='shortest')
40-
score, scores = bleu_scorer.compute_score(option='closest', verbose=1)
40+
score, scores = bleu_scorer.compute_score(option='closest', verbose=verbose)
4141
#score, scores = bleu_scorer.compute_score(option='average', verbose=1)
4242

4343
# return (bleu, bleu_info)

0 commit comments

Comments
 (0)