如何保存Gensim LDA模型列表？

Question

我在一个列表中存储了不同的LDA模型（在相同的文本上，但是都带有不同的#topics）。现在，我要将包含所有模型的列表保存到磁盘中。但是，我不确定这是如何工作的。我应该将其视为列表还是LDA模型？

在gensim website上，我找到了以下代码：

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

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

Answer 1

说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函数，但是如果有更有效的方法可以做到这一点，我也想知道。

Answer 2

就我认为最大的循环效率而言，您可以检出pickle来保存模型或sklearn joblib。 ：）