文本聚类的主题建模效率低下

时间:2018-03-20 09:17:01

标签: python cluster-analysis gensim lda

我尝试使用LDA进行文本聚类,但它没有给我不同的聚类。以下是我的代码

#Import libraries
from gensim import corpora, models
import pandas as pd
from gensim.parsing.preprocessing import STOPWORDS
from itertools import chain

#stop words
stoplist = list(STOPWORDS)
new = ['education','certification','certificate','certified']
stoplist.extend(new)
stoplist.sort()

#read data
dat = pd.read_csv('D:\data_800k.csv',encoding='latin').Certi.tolist()
#remove stop words
texts = [[word for word in document.lower().split() if word not in stoplist] for document in dat]
#dictionary
dictionary = corpora.Dictionary(texts)
#corpus
corpus = [dictionary.doc2bow(text) for text in texts]
#train model
lda = models.LdaMulticore(corpus, id2word=dictionary, num_topics=25, workers=4,minimum_probability=0)
#print topics
lda.print_topics(num_topics=25, num_words=7)
#get corpus
lda_corpus = lda[corpus]
#calculate cutoff score
scores = list(chain(*[[score for topic_id,score in topic] \
                      for topic in [doc for doc in lda_corpus]]))


#threshold
threshold = sum(scores)/len(scores)
threshold
**0.039999999971137644**

#cluster1
cluster1 = [j for i,j in zip(lda_corpus,dat) if i[0][1] > threshold]

#cluster2
cluster2 = [j for i,j in zip(lda_corpus,dat) if i[1][1] > threshold]

问题是cluster1中存在重叠元素,这些元素往往存在于cluster2中,依此类推。

我还尝试手动将阈值提高到0.5,但它给了我同样的问题

1 个答案:

答案 0 :(得分:0)

这是切合实际的。

文档或单词通常都不能唯一地分配给单个群集。

如果您手动标记某些数据,您还会快速找到一些无法明确标记为其中一种的文档。所以它我算法并没有假装有一个很好的独特作业。

相关问题