2013-08-26 52 views
3

我想有一個教程在LSA在這個鏈接工作(編輯:2017年七月刪除死鏈接)潛在語義分析(LSA)教程

下面是本教程的代碼:

titles = [doc1,doc2] 
stopwords = ['and','edition','for','in','little','of','the','to'] 
ignorechars = ''',:'!''' 

class LSA(object): 
    def __init__(self, stopwords, ignorechars): 
     self.stopwords = open('stop words.txt', 'r').read() 
     self.ignorechars = ignorechars 
     self.wdict = {} 
     self.dcount = 0   
    def parse(self, doc): 
     words = doc.split(); 
     for w in words: 
      w = w.lower() 
      if w in self.stopwords: 
       continue 
      elif w in self.wdict: 
       self.wdict[w].append(self.dcount) 
      else: 
       self.wdict[w] = [self.dcount] 
     self.dcount += 1  
    def build(self): 
     self.keys = [k for k in self.wdict.keys() if len(self.wdict[k]) > 1] 
     self.keys.sort() 
     self.A = zeros([len(self.keys), self.dcount]) 
     for i, k in enumerate(self.keys): 
      for d in self.wdict[k]: 
       self.A[i,d] += 1 
    def calc(self): 
     self.U, self.S, self.Vt = svd(self.A) 
    def TFIDF(self): 
     WordsPerDoc = sum(self.A, axis=0)   
     DocsPerWord = sum(asarray(self.A > 0, 'i'), axis=1) 
     rows, cols = self.A.shape 
     for i in range(rows): 
      for j in range(cols): 
       self.A[i,j] = (self.A[i,j]/WordsPerDoc[j]) * log(float(cols)/DocsPerWord[i]) 
    def printA(self): 
     print 'Here is the count matrix' 
     print self.A 
    def printSVD(self): 
     print 'Here are the singular values' 
     print self.S 
     print 'Here are the first 3 columns of the U matrix' 
     print -1*self.U[:, 0:3] 
     print 'Here are the first 3 rows of the Vt matrix' 
     print -1*self.Vt[0:3, :] 

mylsa = LSA(stopwords, ignorechars) 
for t in titles: 
    mylsa.parse(t) 
mylsa.build() 
mylsa.printA() 
mylsa.calc() 
mylsa.printSVD() 

我閱讀並再次閱讀,但我無法想象的東西。如果我執行代碼,結果將如下

Here are the singular values 
[ 4.28485706e+01 3.36652135e-14] 
Here are the first 3 columns of the U matrix 
[[ 3.30049181e-02 -9.99311821e-01 7.14336493e-04] 
[ 6.60098362e-02 1.43697129e-03 6.53394384e-02] 
[ 6.60098362e-02 1.43697129e-03 -9.95952378e-01] 
..., 
[ 3.30049181e-02 7.18485644e-04 2.02381089e-03] 
[ 9.90147543e-02 6.81929920e-03 6.35728804e-03] 
[ 3.30049181e-02 7.18485644e-04 2.02381089e-03]] 
Here are the first 3 rows of the Vt matrix 
array([[ 0.5015178 , 0.86514732], 
    [-0.86514732, 0.5015178 ]]) 

如何從這些矩陣中計算doc1和doc2的相似度?在我自己編寫的tfidf算法中,我得到了一個簡單的浮點數,這裏有3個矩陣。有什麼建議?

+0

教程鏈接是去一些垃圾郵件網站。 –

+0

刪除它。看起來它現在已經死了 – Tasos

回答

0

一種選擇是運行兩個矩陣之間的餘弦相似度。我想你會發現我前段時間發佈的有用信息。我也發佈了這個問題的答案,我發現其他人也給出了很好的答案。

Python: tf-idf-cosine: to find document similarity