Fix to blau score (#165)

* nltk * latest nltk * chlog Co-authored-by: Jirka Borovec <[email protected]>
Lightning-AI · Apr 9, 2021 · a17ccb2 · a17ccb2
1 parent 935994c
commit a17ccb2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -47,6 +47,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 - Fixed when `_stable_1d_sort` to work when n >= N ([PL^6177](https://github.com/PyTorchLightning/pytorch-lightning/pull/6177))
 - Fixed `_computed` attribute not being correctly reset ([#147](https://github.com/PyTorchLightning/metrics/pull/147))
+- Fixed to blau score ([#165](https://github.com/PyTorchLightning/metrics/pull/165))
 
 
 ## [0.2.0] - 2021-03-12

diff --git a/requirements/test.txt b/requirements/test.txt
@@ -14,4 +14,4 @@ yapf>=0.29.0
 cloudpickle>=1.3
 scikit-learn>0.22.1
 scikit-image>0.17.1
-nltk>=3.3
+nltk>=3.6
diff --git a/torchmetrics/functional/nlp.py b/torchmetrics/functional/nlp.py
@@ -101,6 +101,7 @@ def bleu_score(
 
     if smooth:
         precision_scores = torch.add(numerator, torch.ones(n_gram)) / torch.add(denominator, torch.ones(n_gram))
+        precision_scores[0] = numerator[0] / denominator[0]
     else:
         precision_scores = numerator / denominator