Как сохранить список моделей Gensim LDA?

Question

У меня разные модели LDA (с одним и тем же текстом, но с разными # темами), хранящиеся в одном списке.Теперь я хочу сохранить этот список со всеми моделями в нем на моем диске.Однако я не уверен, как это работает.Должен ли я рассматривать это как список или как модель LDA?

На веб-сайте gensim я нашел следующий код:

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>]

Вы можете сохранить несколько файлов одновременно с циклом и счетчиком:

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

Я неЯ знаю о функции Gensim, которая делает это, но если есть более эффективный способ сделать это, я бы тоже хотел знать.

Answer 2

Вы можете проверить pickle , чтобы сохранить модели, или sklearn joblib , насколько я думаю, что эффективность цикла самая большая. :)