inferencing with top k 2

tensorlayer · Sep 6, 2017 · d8375cb · d8375cb
1 parent 3c9837a
commit d8375cb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -34,7 +34,7 @@ Query > donald trump won last nights presidential debate according to snap onlin
  > i think he was a racist
  > he is not a racist
  > he is a liar
- > i dont think he is a disgrace
+ > trump needs to be president
 
 ```
 <!---

diff --git a/main_simple_seq2seq.py b/main_simple_seq2seq.py
@@ -213,7 +213,7 @@ def model(encode_seqs, decode_seqs, is_train=True, reuse=False):
                         o, state = sess.run([y, net_rnn.final_state_decode],
                                         {net_rnn.initial_state_decode: state,
                                         decode_seqs2: [[w_id]]})
-                        w_id = tl.nlp.sample_top(o[0], top_k=3)
+                        w_id = tl.nlp.sample_top(o[0], top_k=2)
                         w = idx2w[w_id]
                         if w_id == end_id:
                             break