LDA主题模型简介及Python实现5 a% M* ]4 U7 ~. c1 R8 G
+ h+ c. B8 i) x% r
一、LDA主题模型简介6 o) M' N. \) g4 h. O) Y
LDA主题模型主要用于推测文档的主题分布,可以将文档集中每篇文档的主题以概率分布的形式给出根据主题进行主题聚类或文本分类。6 v0 m% O6 K6 R$ u+ L
- _# j" A: }! J, U LDA主题模型不关心文档中单词的顺序,通常使用词袋特征(bag-of-word feature)来代表文档。词袋模型介绍可以参考这篇文章:文本向量化表示——词袋模型 - 知乎 / v( y2 M! S: t& A' U6 ]3 S$ u6 {3 F0 `, I8 p) z/ A' J) i6 D( K
了解LDA模型,我们需要先了解LDA的生成模型,LDA认为一篇文章是怎么形成的呢?* _) c+ a% u) O2 O
; |/ r( c' s+ p3 V& q7 f
LDA模型认为主题可以由一个词汇分布来表示,而文章可以由主题分布来表示。 5 q) V5 F4 Y- Q# g# M" Z 8 B4 x' k: v2 B D 比如有两个主题,美食和美妆。LDA说两个主题可以由词汇分布表示,他们分别是:4 ~4 U9 b" \/ }# A# j* z
9 M- R; B. U; I$ N
{面包:0.4,火锅:0.5,眉笔:0.03,腮红:0.07}' L/ M! b0 w! N' m4 h% n
{眉笔:0.4,腮红:0.5,面包:0.03,火锅:0.07}9 O' G ]- A* f0 y4 w
6 v$ T3 v9 ~! o$ \2 a" L 同样,对于两篇文章,LDA认为文章可以由主题分布这么表示: 9 s& |' X) Q( A( v* G6 l . L. A" F$ j% B3 o; y) x5 j3 o* x2 k《美妆日记》{美妆:0.8,美食:0.1,其他:0.1}! k( K' p- G2 F
( e" V0 l& Q r3 x
《美食探索》{美食:0.8,美妆:0.1,其他:0.1} 2 i0 I* R a K6 D0 l7 v 1 e# n- s0 X( s3 c, W& t. ] 所以想要生成一篇文章,可以先以一定的概率选取上述某个主题,再以一定的概率选取那个主题下的某个单词,不断重复这两步就可以生成最终文章。 3 {' h9 Y$ z x6 C+ m2 ` q" x3 c; `# n( o. R, s5 M7 Y7 V
在LDA模型中,一篇文档生成的方式如下:( t3 t/ q/ z/ U4 B
* S2 M) C& t/ L4 T) l' B; ^ ; \$ B! h3 Y e4 q; K5 Y3 b6 K0 t0 B- j3 |4 T
其中,类似Beta分布是二项式分布的共轭先验概率分布,而狄利克雷分布(Dirichlet分布)是多项式分布的共轭先验概率分布。& y1 Q0 {1 S7 `, l8 u: b% F1 V
! }5 x6 {& w$ n$ s6 d* J! Y1 H o3 h: S( f+ S2 m$ A6 t1 y; {1 x# u
5 b8 z3 k# {9 N& y% C' V0 u8 o
如果我们要生成一篇文档,它里面的每个词语出现的概率为:- x) N( T9 w% b, |* ~5 h
2 h) t5 l6 u, Z4 n5 d + E( F6 t6 A' o& R , D; _/ ]; J9 a# j$ v1 O5 K 更详细的数学推导可以见:通俗理解LDA主题模型_结构之法 算法之道-CSDN博客_lda模型 0 u6 ]# w4 B; V5 e; M+ x3 H0 @* s# P3 S N( d
看到文章推断其隐藏的主题分布,就是建模的目的。换言之,人类根据文档生成模型写成了各类文章,然后丢给了计算机,相当于计算机看到的是一篇篇已经写好的文章。现在计算机需要根据一篇篇文章中看到的一系列词归纳出当篇文章的主题,进而得出各个主题各自不同的出现概率:主题分布。 $ F8 J% I& W s7 F5 E7 R 7 Y2 M+ M4 C- @2 } 至于LDA主题模型在计算机中具体是怎么实现的,我们也可以不必细究,现在有很多可以直接用来进行LDA主题分析的包,我们直接用就行。(没错,我就是调包侠) ' {/ Q, d: `) p/ ]( g" o. ^- S: ^* b" W8 Z! I
二、Python实现 8 D; J. f% Y6 b, g 在用Python进行LDA主题模型分析之前,我先对文档进行了分词和去停用词处理(详情可以看我之前的文章:用python对单一微博文档进行分词——jieba分词(加保留词和停用词)_阿丢是丢心心的博客-CSDN博客_jieba 停用词)0 O: o8 `! }9 g I' [
W& w! l$ Q/ v& |' l+ o
我下面的输入文件也是已经分好词的文件0 T) {& l" y' T, [% u8 {" f
8 P+ T) V/ L, [3 m9 e+ c" g7 h# Z+ p
1.导入算法包" v# \9 c4 w2 K
import gensim8 v( l% {# C- M/ y
from gensim import corpora q. {# p4 a6 E4 j. Rimport matplotlib.pyplot as plt ) {6 @& }& B6 @ y0 H# eimport matplotlib " Y7 E: R; u- N" }6 E% V) V1 |8 ~import numpy as np - T2 u# y3 l7 ]& j) {+ Rimport warnings6 M6 `6 ^% B0 d% C7 p( E+ }
warnings.filterwarnings('ignore') # To ignore all warnings that arise here to enhance clarity. f; I6 L- i- c$ Z/ b
& F, x: ?3 m* P6 h2 o) B
from gensim.models.coherencemodel import CoherenceModel $ H" L* a" \% u1 q8 Gfrom gensim.models.ldamodel import LdaModel 8 O! _3 X! H$ X( k* _5 y2.加载数据" S7 [ @1 ]+ `) e" e" U4 _0 {, _
先将文档转化为一个二元列表,其中每个子列表代表一条微博: * Y8 s5 q) ~$ l! m' v8 f+ k* J* {% j+ o4 p6 p
PATH = "E:/data/output.csv" 2 g3 P( |# p% F9 h! q2 E4 e" {/ E+ c2 w) R! E( Z1 i
file_object2=open(PATH,encoding = 'utf-8',errors = 'ignore').read().split('\n') #一行行的读取内容 / J( A! k( l! i3 Hdata_set=[] #建立存储分词的列表$ ~' }& ]2 Y N$ j7 q- ~& \* f, q; |
for i in range(len(file_object2)):1 v) F3 `4 `: H1 Z: @' e9 _
result=[]5 C5 o1 X+ o3 f& A2 ]( L3 |
seg_list = file_object2.split()( }" N' O. m" C0 g+ N
for w in seg_list : #读取每一行分词% f/ ~2 F1 D0 C: [- a4 {
result.append(w) + c% v% ]6 c0 Z0 w. y data_set.append(result) ( j% R, V, d; i) n4 y p" `- Aprint(data_set)4 v% A/ |3 Q; U
构建词典,语料向量化表示: 6 S) c0 k( {; R + ~6 \' W) ]7 C( ]! k$ {dictionary = corpora.Dictionary(data_set) # 构建词典 . O2 [/ L$ e vcorpus = [dictionary.doc2bow(text) for text in data_set] #表示为第几个单词出现了几次4 _2 I" M- l, V L: v
3.构建LDA模型 * U( G& q6 ^7 ~' ^( h+ z$ q8 Aldamodel = LdaModel(corpus, num_topics=10, id2word = dictionary, passes=30,random_state = 1) #分为10个主题 ' \ G0 Z6 P/ ]2 G: D, _% Z7 W$ Zprint(ldamodel.print_topics(num_topics=num_topics, num_words=15)) #每个主题输出15个单词) I* L- D$ K2 ~9 {
这是确定主题数时LDA模型的构建方法,一般我们可以用指标来评估模型好坏,也可以用这些指标来确定最优主题数。一般用来评价LDA主题模型的指标有困惑度(perplexity)和主题一致性(coherence),困惑度越低或者一致性越高说明模型越好。一些研究表明perplexity并不是一个好的指标,所以一般我用coherence来评价模型并选择最优主题,但下面代码两种方法我都用了。 1 E0 t! }: j/ m% R5 \+ A; M* d/ E# W; ^1 d; q
#计算困惑度 4 [+ _1 j. p9 s( jdef perplexity(num_topics): % o. J% |1 P" z/ Z2 b! [6 n% `7 O ldamodel = LdaModel(corpus, num_topics=num_topics, id2word = dictionary, passes=30) ; F4 l2 V K7 e5 W! l print(ldamodel.print_topics(num_topics=num_topics, num_words=15))7 D5 s2 s* ? R4 M y: ?# M
print(ldamodel.log_perplexity(corpus))9 A* P; ]: a8 L2 i6 V8 ]
return ldamodel.log_perplexity(corpus) ) J, X7 k7 s& Z. B( F/ G#计算coherence # E5 J7 E7 o0 z1 g- ~7 Tdef coherence(num_topics): 5 y* _$ M" Z/ L1 O( K ldamodel = LdaModel(corpus, num_topics=num_topics, id2word = dictionary, passes=30,random_state = 1)3 [3 }$ {8 }& R* M( U
print(ldamodel.print_topics(num_topics=num_topics, num_words=10))% ^3 b3 e; e+ u
ldacm = CoherenceModel(model=ldamodel, texts=data_set, dictionary=dictionary, coherence='c_v') . t! E& i7 c" P print(ldacm.get_coherence()) 5 n" Z- n- t+ N8 p$ A return ldacm.get_coherence() * R) n; i/ B, v* t( Y* K4.绘制主题-coherence曲线,选择最佳主题数 - t* Q5 M5 q% M* n" ?2 L5 Dx = range(1,15)# g m- p+ `: J7 S0 A0 i7 ]
# z = [perplexity(i) for i in x] #如果想用困惑度就选这个$ T' K5 y; b, ~! r
y = [coherence(i) for i in x] 3 e. w- H& N0 U' j, `; Splt.plot(x, y) , D4 v7 s5 j) b' d( ~0 z: M2 A# J4 dplt.xlabel('主题数目'), L' S/ U, w" ^- U4 w, j T5 t# h
plt.ylabel('coherence大小')0 z( F0 p, c, U' I1 r# O
plt.rcParams['font.sans-serif']=['SimHei'] U$ O1 @" b' J3 mmatplotlib.rcParams['axes.unicode_minus']=False + q: Z; M: [8 u8 b E4 F6 pplt.title('主题-coherence变化情况')$ o2 M7 D j2 I
plt.show() y! E& y* r1 L- S' @2 u) a 最终能得到各主题的词语分布和这样的图形:, \* L- f, J, N$ h& y. Z1 O
) P' G( H+ {2 A$ G
+ {* o0 z4 D* T' x K+ X - C& w4 U- R0 ]& k 5.结果输出与可视化 $ L/ H w G9 D% L1 `# P" u 通过上述主题评估,我们发现可以选择5作为主题个数,接下来我们可以再跑一次模型,设定主题数为5,并输出每个文档最有可能对应的主题 " S% w+ H1 W: ~3 | 0 [: L' v9 s6 q# Y( a, p5 K' _% hfrom gensim.models import LdaModel% v9 j; _9 P1 h9 c- a
import pandas as pd9 ]0 A% z' e( Z- z* w" q, _; O
from gensim.corpora import Dictionary0 K9 w' a. u8 T
from gensim import corpora, models* w% m+ g1 Z3 }( ~9 A; w; P6 [0 r$ R
import csv5 U E4 r0 b1 I0 L5 `9 I
2 Q @4 U" b, b- H [' ~
# 准备数据( X' ?: x! p( B
PATH = "E:/data/output1.csv" ! U2 h0 A# {8 d" I+ S 5 F5 T9 M( s. c2 `6 {file_object2=open(PATH,encoding = 'utf-8',errors = 'ignore').read().split('\n') #一行行的读取内容4 @( U- A) W" E, X5 ]8 j8 ^' S
data_set=[] #建立存储分词的列表6 x* h; O$ g1 D' o
for i in range(len(file_object2)):5 }. W1 W' F& n- a( A
result=[]: u, r* A: f0 d) g2 C0 C
seg_list = file_object2.split() : U2 l# V/ d1 I2 l6 K$ l for w in seg_list :#读取每一行分词+ b' t+ m" P7 Z8 `4 E, E
result.append(w) " B$ {. U; r9 n* D+ W data_set.append(result)* Z: _: e( j& G. [* }
% e1 b9 b. Y% h( a3 I
dictionary = corpora.Dictionary(data_set) # 构建词典- h9 V- p# Y" M( @& D0 _5 d* k7 R
corpus = [dictionary.doc2bow(text) for text in data_set]/ s: y7 I: i. M8 z) m2 `3 A! i
$ j% I0 b8 z. {( m- Hlda = LdaModel(corpus=corpus, id2word=dictionary, num_topics=5, passes = 30,random_state=1)( F1 h1 Q2 F9 V8 @- C
topic_list=lda.print_topics()" ` N1 E& q9 ]
print(topic_list)7 c, e$ p) }) m3 ^
1 w7 m) K, a1 x. t
for i in lda.get_document_topics(corpus)[:]:* m( T" h" l4 T& b7 G" L
listj=[]0 D' Y: b$ E% v" Q s9 d; Y1 |9 P
for j in i: 2 Y9 ]/ d$ S3 t listj.append(j[1])! v. I0 D/ B# X7 @
bz=listj.index(max(listj)) ( k2 n& R. `0 \9 n0 k print(i[bz][0]) ; |- j" K& J* m* T- O 9 \/ t4 ?0 x+ [3 t% o, i 同时我们可以用pyLDAvis对LDA模型结果进行可视化: # E+ u7 k# T( U" _3 E( V# x. E x- M
import pyLDAvis.gensim7 V9 o( K5 x( f4 Y* x# D/ ~# p& B4 z8 I
pyLDAvis.enable_notebook(). {! y$ M# e; j0 c
data = pyLDAvis.gensim.prepare(lda, corpus, dictionary) 2 G# T$ p3 n" i, q3 PpyLDAvis.save_html(data, 'E:/data/3topic.html') / U- O; d- N1 K( J; x 大概能得到这样的结果: 0 y" g0 p+ s4 `5 B; E - a& U; K" \" J, Z4 P2 F. Q1 }; F2 }9 c Y* K
) _8 Q |. Z! w+ T7 g 左侧圆圈表示主题,右侧表示各个词语对主题的贡献度。5 d5 n3 ~/ y- V, K* X6 Y
8 o) V: r1 C3 n$ b q7 n* I# C所有代码如下: 1 z$ R o0 E+ i Y( Dimport gensim$ G' S% S/ g# ?$ Y! J, t
from gensim import corpora + _& r( o. \) z/ o! \% Uimport matplotlib.pyplot as plt9 s5 d8 G4 b( [4 V/ }5 Z
import matplotlib # v i9 {6 r+ {, F+ |: J% \& }5 ximport numpy as np7 Y2 w- H' }$ x9 q3 ]
import warnings1 N& }; o Q; q$ q2 u W$ \
warnings.filterwarnings('ignore') # To ignore all warnings that arise here to enhance clarity; d ~2 c8 \' k5 b: n
' X7 n% [7 _' \+ _ q1 T
from gensim.models.coherencemodel import CoherenceModel 2 u. f3 D2 J' K: ]. s2 p$ S) xfrom gensim.models.ldamodel import LdaModel3 b- i) C2 P! K3 x* A- G
1 u+ k0 e, {$ C6 s9 g6 c* J # D/ q7 Y% D3 D( n: m/ U3 f9 {( b; ]2 q* q3 R
# 准备数据 2 P/ z4 k- M) J3 b) S' ^* t; }PATH = "E:/data/output.csv" + ^1 c% F* {% w p- [1 i6 \4 m" A# e2 f
file_object2=open(PATH,encoding = 'utf-8',errors = 'ignore').read().split('\n') #一行行的读取内容 p' M: Y0 P' |data_set=[] #建立存储分词的列表) |. ` j4 e* ^! ^& \, |
for i in range(len(file_object2)): - L5 p7 y. {! p Q result=[]+ K, u, F& }9 q: K( G( \
seg_list = file_object2.split()7 G# K) R; {, W0 C# h
for w in seg_list :#读取每一行分词/ e3 |* u$ N" L% y% ~, t/ ]% T
result.append(w) & u5 W1 o3 g3 R/ f/ n( M0 U# } j data_set.append(result) ; F8 d6 s! U& t( C$ D5 y# `print(data_set)3 |& C& C& y9 {
* f5 r. J" O" E, z) \ ) n; k; b- w# b! ~9 Jdictionary = corpora.Dictionary(data_set) # 构建 document-term matrix0 Z/ H# W/ |! M7 y' o5 v: m. M* E5 `
corpus = [dictionary.doc2bow(text) for text in data_set] 9 e c; {3 h! |) O#Lda = gensim.models.ldamodel.LdaModel # 创建LDA对象 % ^9 A+ h @2 S4 W; m+ E. I$ k) ]: V2 k' P+ O+ G4 M m
#计算困惑度 , l( i) Q, s' l/ A# q% {4 f7 l0 E4 e4 Fdef perplexity(num_topics): $ E" D! R6 }4 X+ ?7 ~. ~ ldamodel = LdaModel(corpus, num_topics=num_topics, id2word = dictionary, passes=30). [% {/ S1 O- h
print(ldamodel.print_topics(num_topics=num_topics, num_words=15)) ! d3 u, @1 y' d0 ^ print(ldamodel.log_perplexity(corpus)) C& O2 @) v; L return ldamodel.log_perplexity(corpus)( X6 D$ l# @4 ^' n7 q
+ k) J+ D$ o y& @" o9 x2 G
#计算coherence3 N" @% c* z/ D) v
def coherence(num_topics):" T! L( r8 |1 F
ldamodel = LdaModel(corpus, num_topics=num_topics, id2word = dictionary, passes=30,random_state = 1) - N3 g- @" U5 u% E* ^ print(ldamodel.print_topics(num_topics=num_topics, num_words=10)) ) o/ r% ]( G* K9 s, ?$ W* \" H ldacm = CoherenceModel(model=ldamodel, texts=data_set, dictionary=dictionary, coherence='c_v') 1 ~$ |2 U( w5 q, A* Y* N7 @4 F5 e print(ldacm.get_coherence()) " @4 S: Y5 u$ | return ldacm.get_coherence()' D- H% _( d" A! T+ K/ ]4 {% }
" B4 m- V4 v, N$ i2 e# 绘制困惑度折线图1 n# J6 W ^/ E" H
x = range(1,15) @+ y, L3 s9 k) E# x# z = [perplexity(i) for i in x]; z( M' A3 k Y- G2 [, X1 G
y = [coherence(i) for i in x] # F) Y4 `5 _# @# iplt.plot(x, y) ' z/ \6 t% N; d) Dplt.xlabel('主题数目')5 O/ M* @% n3 q* k3 }! R( n
plt.ylabel('coherence大小') 6 b. I% |# }( S2 `; ~/ E5 s0 oplt.rcParams['font.sans-serif']=['SimHei'] / \' V9 \1 G' P4 ^; d& `3 K1 n! tmatplotlib.rcParams['axes.unicode_minus']=False " b0 }. K2 }) N6 }6 o0 Xplt.title('主题-coherence变化情况')6 {% n L" o# w
plt.show()7 D5 t/ K. K$ {' U/ _2 M+ |
3 C5 \, O2 u) f3 ]5 b9 s
from gensim.models import LdaModel5 O" l3 e$ P: f% k7 _! x8 K
import pandas as pd1 ]9 ^1 x. ^, i8 S' b3 Q, L& R
from gensim.corpora import Dictionary1 U" \6 m2 J4 J. u3 P
from gensim import corpora, models% `9 q1 @ ^ w( f6 Y
import csv9 B' m Q! a: ]) { E) g- ?
6 I( k2 Y* n/ u j5 @6 d
# 准备数据0 V/ i5 T: b7 C
PATH = "E:/data/output1.csv" ! Z' r/ d' h9 C6 _1 s, E 7 A1 _7 A" ] Vfile_object2=open(PATH,encoding = 'utf-8',errors = 'ignore').read().split('\n') #一行行的读取内容 ! u) Q7 |7 m) }" Idata_set=[] #建立存储分词的列表4 W7 N! h0 G( |2 n
for i in range(len(file_object2)): $ u- p4 d: \: P$ f. N result=[]! b, M& q- H. m% G* l, J/ g
seg_list = file_object2.split() 2 e Q# W" w' W7 Z; H for w in seg_list :#读取每一行分词% w: @- n. ?" C; b
result.append(w)9 Y/ t, X$ P$ z! T6 x
data_set.append(result)& m1 P/ L1 o; |# }3 h
2 J3 s3 v; M. c
dictionary = corpora.Dictionary(data_set) # 构建 document-term matrix + f6 L f( u- N* |. ]corpus = [dictionary.doc2bow(text) for text in data_set]" _* f: ~' R" E) P$ m9 |
+ E. p. Q* K/ v0 D/ I5 U) H
lda = LdaModel(corpus=corpus, id2word=dictionary, num_topics=5, passes = 30,random_state=1) % J3 I4 Z+ |3 D1 g, c3 @/ Htopic_list=lda.print_topics()& [* Y; a, S! |7 u" x! Z. G: b/ ^
print(topic_list)! Q9 w# K5 c. i) \: B, E; U
+ b1 }; E9 f! `0 e+ B$ c: R8 ^7 [result_list =[] * e7 i* V q9 F1 n7 j" gfor i in lda.get_document_topics(corpus)[:]: 3 e; i' w- { ~& x$ i# { listj=[] ( N E: ^1 i# m, C- E; q for j in i: ! }) J a6 _( \. u& O% G0 D$ ] listj.append(j[1])4 G, {) J% U; O
bz=listj.index(max(listj)), m! s3 J/ ^7 r: ]- T( K* X
result_list.append(i[bz][0])8 J/ B; r9 X2 o3 N9 Y% R# L
print(result_list)& Z0 X3 L' s/ l0 Z( k