def testBleuScorerBrevityPenalty(self):
scorer = scorers.BleuScorer(max_ngram=4)
scorer.AddSentence('1 2 3 4 5', '1 2 3 4 -2')
# No penalty.
expected_score = (4 / 5 * 3 / 4 * 2 / 3 * 1 / 2)**(1 / 4)
self.assertAlmostEqual(0.6687403, expected_score)
self.assertAlmostEqual(expected_score, scorer.ComputeOverallScore())
scorer = scorers.BleuScorer(max_ngram=4)
scorer.AddSentence('1 2 3 4 5', '1 2 3 4')
expected_score = math.exp(1 - 5 / 4) * (4 / 4 * 3 / 3 * 2 / 2 * 1 /
1)**(1 / 4)
self.assertAlmostEqual(0.7788008, expected_score)
self.assertAlmostEqual(expected_score, scorer.ComputeOverallScore())
def testBleuScorerDocTest(self):
scorer = scorers.BleuScorer(max_ngram=4)
scorer.AddSentence('hyp matches ref str', 'hyp matches ref str')
self.assertAlmostEqual((4 / 4 * 3 / 3 * 2 / 2 * 1 / 1)**(1 / 4),
scorer.ComputeOverallScore())
scorer.AddSentence('almost right', 'almost write')
self.assertAlmostEqual((5 / 6 * 3 / 4 * 2 / 2 * 1 / 1)**(1 / 4),
scorer.ComputeOverallScore())
def testBleuScorerMatchesCppImplementation(self):
filename = test_helper.test_src_dir_path(
'core/ops/testdata/wmt/sm18.txt')
scorer = scorers.BleuScorer()
with open(filename, 'rb') as fp:
for line in fp:
hyp, ref = line[:-1].split(b'\t')
scorer.AddSentence(ref, hyp)
self.assertAlmostEqual(0.313776,
scorer.ComputeOverallScore(),
places=5)
def testBleuScorerSentencesShorterThanMaxNGram(self):
scorer = scorers.BleuScorer(max_ngram=4)
scorer.AddSentence('', '')
self.assertAlmostEqual(0.0, scorer.ComputeOverallScore())
scorer.AddSentence('a', 'a')
self.assertAlmostEqual(1.0, scorer.ComputeOverallScore())
scorer.AddSentence('a b', 'a b')
self.assertAlmostEqual(1.0, scorer.ComputeOverallScore())
scorer.AddSentence('a b c', 'a b c')
self.assertAlmostEqual(1.0, scorer.ComputeOverallScore())
scorer.AddSentence('a b c d', 'a b c d')
self.assertAlmostEqual(1.0, scorer.ComputeOverallScore())
def __init__(self, **kwargs):
self._scorer = scorers.BleuScorer(**kwargs)
def testBleuScorerMatchesCppImplementationOneExample(self):
scorer = scorers.BleuScorer()
scorer.AddSentence(
'It is obvious that this will have a certain influence .',
'It is clear that this will have a certain influence .')
self.assertAlmostEqual(0.74194467, scorer.ComputeOverallScore())
def testBleuScorerClipsExtraHypNGrams(self):
scorer = scorers.BleuScorer(max_ngram=4)
scorer.AddSentence('a b c d', 'a a b c d')
self.assertAlmostEqual((4 / 5 * 3 / 4 * 2 / 3 * 1 / 2)**(1 / 4),
scorer.ComputeOverallScore())