python - 如何在 Gensim 中将模型、字典和语料库保存到磁盘，然后再次加载它们？

Question

在 Gensim 的 documentation ， 它说:

You can save trained models to disk and later load them back, either to continue training on new training documents or to transform new documents.

我想用字典、语料库和 tf.idf 模型来做到这一点。然而，文档似乎说这是可能的，没有解释如何保存这些东西并重新加载它们。

你怎么做到这一点？

我一直在使用泡菜，但不知道这是否正确......

import pickle
pickle.dump(tfidf, open("tfidf.p", "wb"))
tfidf_reloaded = pickle.load(open("tfidf.p", "rb"))

Answer 1

将字典和语料库保存到磁盘

dictionary.save(DICT_PATH)
corpora.MmCorpus.serialize(CORPUS_PATH, corpus)

从磁盘加载字典和语料库

loaded_dict = corpora.Dictionary.load(DICT_PATH)
loaded_corp = corpora.MmCorpus(CORPUS_PATH)