CODEWITHSUNDEEP
tensorflow
IH_K
IH_K

Reputation: 71

tensorflow seq2seq with multiple outputs

I just started to work on tensorflow not so long. I'm working on the seq2seq model and using seq2seq example code.

I want to modify seq2seq model code to get top-k outputs (k is 5 or 10) for the Reinforcement learning model, not to get top-1 output.

First, I think I should modify decoder part of the seq2seq somehow, but I don't know which part is to change.

Is there any references or codes for the problem?

Upvotes: 0

Views: 911

Answers (2)

lifang
lifang

Reputation: 1715

tf.contrib.seq2seq.BeamSearchDecoder would do the magic for you.

Upvotes: 1

Tilney
Tilney

Reputation: 318

check out https://github.com/tensorflow/tensorflow/issues/654. There are some discussions on this, but no worked example yet.

Upvotes: 1

Related Questions