有没有一种方法可以控制序列的输出长度以实现序列文本汇总模型?

时间:2019-01-14 20:13:59

标签: python text summary seq2seq

是否有一种方法可以控制用于文本汇总的s​​eq2seq模型产生的单词或字符的数量?例子:

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

1个输出:我的狗快,爱拥抱。

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

1 个答案:

答案 0 :(得分:1)

这可以通过产生输出的解码器步骤数来控制,在此repo中,有多种方法可以进行文本汇总,而在每种方法中,都有一个参数,例如在{{ 3}},有一个名为 max_dec_steps 的参数(在单元格28中),该参数控制解码器的最大时间步长(最大摘要令牌),该参数仅反映输出语句的长度,回购记录员还解释说model中将详细介绍其他多个模型。

希望这会有所帮助