时间戳

首页 > 解决方案 > 如何保存 Gensim LDA 模型列表？

python - 如何保存 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 模型，不适用于具有多个模型的列表。保存我的模型列表的最佳方法是什么？

标签： pythonlistsavegensim

解决方案

说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 函数，但如果有更有效的方法来执行此操作，我也很想知道。

推荐阅读