有没有一种方法可以控制序列的输出长度以对文本摘要模型进行排序？

Posted 2023-01-23

是否有一种方法可以控制用于文本摘要的seq2seq模型产生的单词或字符的数量？示例：

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

1输出：我的狗快，爱拥抱。

2输出：我的狗是最快的狗，也喜欢拥抱。

答案

这可以通过产生输出的解码器步骤数来控制，在此repo中，有多种方法可以进行文本汇总，而在每种方法中，都有一个参数，例如在model的情况下，是一个名为max_dec_steps的参数（在单元格28中），它控制解码器的最大时间步长（最大摘要令牌），该时间步长仅反映输出语句的长度，在此[repo writer]还详细说明了其他多个模型。 C0]。

希望这会有所帮助