fixed small bug in bleu computation

a3932b0d · Christian Fuß · 375b84d6 · a3932b0d
Commit a3932b0d authored Nov 12, 2019 by Christian Fuß
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/main/resources/templates/gluon/CNNSupervisedTrainer.ftl src/main/resources/templates/gluon/CNNSupervisedTrainer.ftl +2 -1

No files found.
--- a/src/main/resources/templates/gluon/CNNSupervisedTrainer.ftl
+++ b/src/main/resources/templates/gluon/CNNSupervisedTrainer.ftl
@@ -116,7 +116,8 @@ class BLEU(mx.metric.EvalMetric):
                    i *= 2
                    match_counts = 1 / i

-                precisions[n] = match_counts / counts
+                if (match_counts / counts) > 0:
+                    precisions[n] = match_counts / counts

        bleu = self._get_brevity_penalty() * math.exp(sum(map(math.log, precisions)) / self.N)