0
点赞
收藏
分享

微信扫一扫

gensim读取已训练模型LDA模型的模型与dictionary

import pyLDAvis.gensim
from gensim import models, corpora
from gensim.corpora import Dictionary

all_data = [
    ['青绿色', '放', '几天', '塑料袋', '里', '刺鼻', '气味', '药水', '味道', '说', '选择', '应季', '水果', '最佳', '踩', '雷', '水果'],
    ['徒有其表', '皮厚肉', '渣', '一点', '甜', '入口', '评论', '欺骗', '消费者'], ['果子', '全是', '伤', '放', '烂', '几个', '放', '烂', '几个'],
    ['路边', '两块钱', '一斤', '橙子', '没什么', '区别', '水分', '甜度', '称为', '品牌', '京东', '买', '水果', '三思'],
    ['橙子', '酸', '一点', '不值'], ['垃圾', '沒', '水份', '别买'], ['质量', '太差', '坏', '吃'],
    ['杨氏', '真', '买', '砸', '招牌', '皮厚肉', '干麻'],
    ['生鲜', '越来越', '不行', '一点', '品质', '售后服务', '可言'],
    ['买', '箱来', '送', '反馈', '说', '干', '京东', '搞', '飞机', '害人不浅'],
    ['橙子', '品质', '太差', '三个', '霉变'],
    ['新鲜', '橙子', '两头', '干枯', '水分', '新货']
]

# 加载各种模型
lda_model = models.ldamodel.LdaModel.load('../pro1/lda_5.model')  # 加载lda
lda_dictionary = corpora.Dictionary.load('../pro1/lda_5.model.id2word')  # 加载dictionary
text_array = [['volume', 'eventually', 'metric', 'rally']]
lda_corpus = [lda_dictionary.doc2bow(i) for i in all_data]  # 转换dict

之后,就可以使用:
lda_model:lda模型
lda_dictionary:词与id的转换列表
lda_corpus:这个是词库,不会随LDA保存,需要随时读取随时用,其中all_data就是转换为corpus的文本

举报

相关推荐

0 条评论