nlp - 如何使用 GPT-2 进行主题建模？

Question

我想从语料库中生成主题和子主题。如果有人可以分享python代码，那就太好了。

Answer 1

好吧，最简单的方法是使用 hugginface 库。他们已经有用于从多个生成模型生成文本的脚本，包括 gpt-2。您正在寻找的脚本是run_generation.py。您可以提供不同长度的提示，并且可以使用它们生成文本。