tf2-ndg-benchmarks

Neural Dialogue Generation Benchmarks implemented TensorFlow 2.0

Usage

Run under the command for setup.

poetry install

Run under python code for using metrics.

from tf2-ndg-benchmarks import metrics

reference = 'It is a guide to action that ensures that the military will forever heed Party commands'
hypothesis = 'It is a guide to action which ensures that the military always obeys the commands of the party'
bleu = metrics.bleu.Bleu()
score = bleu.sentence_score(reference, hypothesis)