add sampling

README.md

@@ -1,6 +1,4 @@
-# Transformer
-
-
+# WIP
 Implementation of the *Transformer* model in the paper:
 
 > Ashish Vaswani, et al. ["Attention is all you need."](http://papers.nips.cc/paper/7181-attention-is-all-you-need.pdf) NIPS 2017.

eval.py

@@ -6,6 +6,7 @@
         Oct 2018
 """
 import click
+import numpy as np
 from data import DatasetManager, recover_sentence
 from transformer import Transformer
 from nltk.translate.bleu_score import corpus_bleu
@@ -43,6 +44,12 @@ def eval(model_name, file_prefix):
         num_sentences=len(hypos),
         bleu_score=bleu_score * 100.,
     )
+
+    # Sample a few translated sentences.
+    indices = np.random.choice(list(range(len(hypos))), size=10, replace=False)
+    for i in indices:
+        print(f"Source: '{refs[i][0]}' ==> Target: '{hypos[i]}'.")
+
     print(results)