我如何生成句子并给出一些关键词？

Posted 2021-05-03

如何在给出一些无序的关键字列表的情况下生成句子？

例如。给定关键字school，going。生成句子，例如：

• 汤姆和杰里不在雨中去到[[学校。
他们明天
• 去
到学校。为什么他们明天不
• 去
明天去[[学校？保持去到
• 学校
。到学校，他们将是
• 参加
。

答案

我想您想使用model.generate()生成？您可以看到此方法here的所有可能参数。

您可以通过将输入作为input_ids传递给generate()方法来“准备”生成过程，例如以“ Tomorrow it will”开头的句子，以强制模型完成句子。

但是我也想使用关键字:)]这样的功能>