gensim.models.ldamodel.LdaModel.save
时间: 2023-09-18 19:12:11 浏览: 117
`gensim.models.ldamodel.LdaModel.save` 是一个方法,用于将训练好的 LDA 模型保存到磁盘上,以便后续可以重新加载该模型进行使用。该方法的参数是要保存的文件名,通常以 ".lda" 为后缀。例如:
```python
from gensim.models.ldamodel import LdaModel
# 假设已经通过某种方式训练好了 LDA 模型
lda_model = LdaModel(corpus, num_topics=10, id2word=dictionary)
# 将训练好的 LDA 模型保存到磁盘上
lda_model.save('my_lda_model.lda')
```
该代码将训练好的 LDA 模型保存到名为 "my_lda_model.lda" 的文件中。要重新加载该模型,可以使用 `gensim.models.ldamodel.LdaModel.load` 方法。例如:
```python
from gensim.models.ldamodel import LdaModel
# 加载之前保存的 LDA 模型
lda_model = LdaModel.load('my_lda_model.lda')
# 使用加载的 LDA 模型进行推断
doc = ['this', 'is', 'a', 'test', 'document']
doc_bow = lda_model.id2word.doc2bow(doc)
doc_topics = lda_model.get_document_topics(doc_bow)
```
以上代码会将之前保存的 LDA 模型加载到 `lda_model` 变量中,并使用该模型对一个文档进行主题推断。
阅读全文