Updated test

48a98935 · Sebastian N. · c37e1ead · 48a98935
Commit 48a98935 authored Jan 28, 2020 by Sebastian N.
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

src/test/resources/target_code/gluon/CNNSupervisedTrainer_mnist_mnistClassifier_net.py ...e/gluon/CNNSupervisedTrainer_mnist_mnistClassifier_net.py +6 -1

No files found.
--- a/src/test/resources/target_code/gluon/CNNSupervisedTrainer_mnist_mnistClassifier_net.py
+++ b/src/test/resources/target_code/gluon/CNNSupervisedTrainer_mnist_mnistClassifier_net.py
@@ -143,7 +143,12 @@ class BLEU(mx.metric.EvalMetric):
        if self._size_hyp >= self._size_ref:
            return 1
        else:
-            return math.exp(1 - (self._size_ref / self._size_hyp))
+            if self._size_hyp > 0:
+                size_hyp = self._size_hyp
+            else:
+                size_hyp = 1
+
+            return math.exp(1 - (self._size_ref / size_hyp))

    @staticmethod
    def _get_ngrams(sentence, n):