如何保存Gensim LDA型号列表?

2024-04-25 18:07:51 发布

您现在位置:Python中文网/ 问答频道 /正文

我在一个列表中存储了不同的LDA模型(在同一个文本上,但都有不同的主题)。现在,我想把这个列表和所有模型保存到我的磁盘上。不过,我不知道这是怎么回事。我应该将is视为列表还是LDA模型?你知道吗

gensim website上,我发现了以下代码:

from gensim.test.utils import datapath
>>>
>>> # Save model to disk.
>>> temp_file = datapath("model")
>>> lda.save(temp_file)

但是,这适用于单独的LDA模型,而不适用于具有多个模型的列表。保存模型列表的最佳方法是什么?你知道吗


Tags: 代码from模型文本主题列表modelis
2条回答

假设trained_models是您的LDA模型列表:

trained_models = 
[<gensim.models.ldamodel.LdaModel at 0x1f321825668>,
 <gensim.models.ldamodel.LdaModel at 0x1f32181ffd0>]

可以使用for循环和计数器一次保存多个文件:

i = 1
for model in trained_models:
    model.save("model{}.gensim".format(i))
    i += 1

我不知道有哪个gensim函数能做到这一点,但如果有更有效的方法,我也想知道。你知道吗

你可以查看pickle来保存模型或者sklearn joblib,我认为这是最有效的循环效率。:)

相关问题 更多 >