如何保存Gensim LDA型号列表



我有一个不同的LDA模型(在同一文本上,但全部带有不同的#Topics(存储在一个列表中。现在,我想将此列表保存在我的磁盘上。但是,我不确定这是如何工作的。我应该当作列表还是LDA型号?

在Gensim网站上,我找到了以下代码:

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

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

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功能,但是如果有更有效的方法可以做到这一点,我也很想知道。

您可以查看泡菜以保存模型或sklearn joblib,而循环效率我认为这是最重要的。:(

最新更新