1. 程式人生 > >文字向量化---從向量到向量(tfidf)

文字向量化---從向量到向量(tfidf)

corpus = [dictionary.doc2bow(text) for text in texts]

tfidf = models.TfidfModel(corpus) # 第一步--初始化一個模型


doc_bow = [(0, 1), (1, 1)]

print tfidf[doc_bow] # 第二步--用模型轉換向量

 

[(0, 0.70710678), (1, 0.70710678)]


####或者在整個語料上應用轉換

corpus_tfidf = tfidf[corpus]

for doc in corpus_tfidf:

	print doc

 

[(0, 0.57735026918962573), (1, 0.57735026918962573), (2, 0.57735026918962573)]

[(0, 0.44424552527467476), (3, 0.44424552527467476), (4, 0.44424552527467476), (5, 0.32448702061385548), (6, 0.44424552527467476), (7, 0.32448702061385548)]

[(2, 0.5710059809418182), (5, 0.41707573620227772), (7, 0.41707573620227772), (8, 0.5710059809418182)]

[(1, 0.49182558987264147), (5, 0.71848116070837686), (8, 0.49182558987264147)]

[(3, 0.62825804686700459), (6, 0.62825804686700459), (7, 0.45889394536615247)]

[(9, 1.0)]

[(9, 0.70710678118654746), (10, 0.70710678118654746)]

[(9, 0.50804290089167492), (10, 0.50804290089167492), (11, 0.69554641952003704)]

[(4, 0.62825804686700459), (10, 0.45889394536615247), (11, 0.62825804686700459)]

很多模型就是基於tf-idf來做的

比如lsi,lda等

舉個栗子


lsi = models.LsiModel(corpus_tfidf, id2word=dictionary, num_topics=2) # 初始化一個LSI轉換

corpus_lsi = lsi[corpus_tfidf] # 在原始語料上建立一個雙重封裝器: bow->tfidf->fold-in-lsi   

這裡我們用潛在語義索引(Latent Semantic Indexing)將我們的Tf-Idf語料庫轉換到潛在2-D空間(2-D因為我們設定 num_topics=2)。現在你可以覺得奇怪:這兩個潛在維度是什麼?讓我們檢查一下models.LsiModel.print_topics():


lsi.print_topics(2)

 

topic #0(1.594): -0.703*"trees" + -0.538*"graph" + -0.402*"minors" + -0.187*"survey" + -0.061*"system" + -0.060*"response" + -0.060*"time" + -0.058*"user" + -0.049*"computer" + -0.035*"interface"

topic #1(1.476): -0.460*"system" + -0.373*"user" + -0.332*"eps" + -0.328*"interface" + -0.320*"response" + -0.320*"time" + -0.293*"computer" + -0.280*"human" + -0.171*"survey" + 0.161*"trees"  

得到主題-詞分佈


for doc in corpus_lsi: # 在這裡,bow->tfidf 和 tfidf->lsi 轉換實際上都是即時執行的

	print doc

 

[(0, -0.066), (1, 0.520)] # "Human machine interface for lab abc computer applications"

[(0, -0.197), (1, 0.761)] # "A survey of user opinion of computer system response time"

[(0, -0.090), (1, 0.724)] # "The EPS user interface management system"

[(0, -0.076), (1, 0.632)] # "System and human system engineering testing of EPS"

[(0, -0.102), (1, 0.574)] # "Relation of user perceived response time to error measurement"

[(0, -0.703), (1, -0.161)] # "The generation of random binary unordered trees"

[(0, -0.877), (1, -0.168)] # "The intersection graph of paths in trees"

[(0, -0.910), (1, -0.141)] # "Graph minors IV Widths of trees and well quasi ordering"

[(0, -0.617), (1, 0.054)] # "Graph minors A survey"

2個主題,0是第一個topic,1是第二個topic

所以這是文件--主題分佈!!!!