具有多个输出的 TensorFlow seq2seq

Question

我刚开始研究张量流的时间不长。我正在研究 seq2seq 模型并使用 seq2seq 示例代码。

我想修改 seq2seq 模型代码以获得强化学习模型的 top-k 输出（k 为 5 或 10），而不是获得 top-1 输出。

首先，我认为我应该以某种方式修改 seq2seq 的解码器部分，但我不知道要更改哪一部分。

是否有针对该问题的任何参考或代码？

Answer 1

查看 https://github.com/tensorflow/tensorflow/issues/654。对此有一些讨论，但还没有工作的例子。

Answer 2

tf.contrib.seq2seq.BeamSearchDecoder会为你创造魔术。