使用用户提供的关键字生成文本

数据挖掘 深度学习 nlp 语言模型 文本生成 openai-gpt
2022-02-20 16:13:12

我有一个用例,我需要根据一组用户提供的关键字生成句子。这是我需要的一个例子:

用户输入:

最终用户:数据科学家

地区:中东

国家:阿联酋

解决方案:大熊猫

应用:机器学习

好处:降低成本和运行时间

输出(大括号只是为了突出显示):

了解 {中东}(例如 {UAE})的{数据科学家}如何使用 {BigPanda} 来简化他们的{机器学习}流程以{降低成本和运行时间}。


所以模型需要使用用户给出的关键词,生成相似的句子。我还有一个包含大约 2000 个此类句子的数据集,可能会派上用场。

我认为这可能的一种方法是使用 GPT-2 模型,并可能使用数据集对其进行微调,但我无法弄清楚我将如何实际使用它来完成类似的事情。

1个回答

是的,微调 GPT2 可以帮助您完成这个目标。但唯一关心的是您拥有的输入数据的大小。要获得性能更好的模型,您必须拥有更大的输入集。