确保在编解码器文本生成深度学习模型中存在单词/标记/名词

Question

我遇到了一个问题，在解码和生成抽象风格的句子时，我希望确保生成特定的标记/单词。

我正在使用LSTM和transformer模型等深度学习模型来生成短句(100-200个字符)。我希望在生成的文本中出现一些单词，如地点或名词(如品牌名称)。

我不确定是否有关于这方面的研究，在广泛的搜索之后，我真的找不到一篇论文。

提亚，任何线索或建议我们都很感激。:)

Answer 1

我不确定，但你可以试着根据这些特定的单词来调整你的输出。您的训练器可以像一个seq2seq解码器，但它可以关注那些特定的单词，而不是编码器的输出。