Spaces:
Running
Running
Added bleu prefix to return dict's key
#3
by
alvations
- opened
bleu.py
CHANGED
|
@@ -123,11 +123,20 @@ class Bleu(evaluate.Metric):
|
|
| 123 |
reference_corpus=references, translation_corpus=predictions, max_order=max_order, smooth=smooth
|
| 124 |
)
|
| 125 |
(bleu, precisions, bp, ratio, translation_length, reference_length) = score
|
| 126 |
-
|
|
|
|
| 127 |
"bleu": bleu,
|
| 128 |
-
"
|
| 129 |
-
"
|
| 130 |
-
"
|
| 131 |
-
"
|
| 132 |
-
"
|
| 133 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 123 |
reference_corpus=references, translation_corpus=predictions, max_order=max_order, smooth=smooth
|
| 124 |
)
|
| 125 |
(bleu, precisions, bp, ratio, translation_length, reference_length) = score
|
| 126 |
+
|
| 127 |
+
results = {
|
| 128 |
"bleu": bleu,
|
| 129 |
+
"bleu_precisions": precisions,
|
| 130 |
+
"bleu_brevity_penalty": bp,
|
| 131 |
+
"bleu_length_ratio": ratio,
|
| 132 |
+
"bleu_translation_length": translation_length,
|
| 133 |
+
"bleu_reference_length": reference_length,
|
| 134 |
}
|
| 135 |
+
|
| 136 |
+
# Add explicit floats values for precisions,
|
| 137 |
+
# so that tensorboard scalars automatically picks it up.
|
| 138 |
+
for n, p in enumerate(precisions, 1):
|
| 139 |
+
results[f'bleu_{n}gram_precisions'] = p
|
| 140 |
+
|
| 141 |
+
return results
|
| 142 |
+
|