0

有没有办法控制用于文本摘要的 seq2seq 模型产生的单词或字符的数量?例子:

“我的狗是世界上跑得最快的狗。它也喜欢拥抱。”

1 输出:我的狗很快,喜欢拥抱。

2 输出:我的狗是跑得最快的狗,也喜欢拥抱。

4

1 回答 1

1

这可以通过产生输出的解码器步骤的数量来控制,在这个repo中有多种文本摘要方法,其中每个都有一个参数,就像这个模型的情况一样,有一个名为max_dec_steps的参数(在控制解码器的最大时间步长(最大摘要标记)的单元格 28)中,它简单地反映了输出句子的长度,repo 作者还在本博客系列中详细解释了多个其他模型。

希望这会有所帮助

于 2020-01-11T10:03:21.127 回答