1. 程式人生 > >一種根據關鍵字進行分類的文字分類演算法

一種根據關鍵字進行分類的文字分類演算法

這樣我們可以得出這個演算法的重點:

1.提取關鍵字

如何自動提取關鍵字呢?我們知道IDF值在一定程度上可以表達一個詞的重要程度,像“我的”,“你的”這樣的關鍵字肯定無法判斷出文章的類別,他們的IDF值也是較低的。而“AK47”,“火箭炮”這樣的關鍵字可以判斷出文章類別,他們的IDF也比一般的詞要高。所以,我們只需要選擇IDF高於一定值的詞,就可以提取出絕大多數關鍵字了。

2.關鍵字分類

識別關鍵字的類別也是一個大問題,如果手工識別是不現實的,網上也沒有什麼演算法是進行關鍵字分類的。想來想去,最後還是用IDF值把這個問題給解決了。其實思路很簡單,可以說是上面一個問題的一種延續。假設現在我們有軍事,經濟,人文的文字各100篇。將經濟,人文的文章複製4次,這樣,我們就有軍事文字100篇,經濟,人文的文字各500篇(注意,經濟人文的文章都是有重複的,每篇文章存在4個完全相同的副本)。

然後我們計算這些文章的IDF值,想想會出現什麼結果?我們知道IDF的計算公式是log(總文章數/出現次數)。假設關鍵字“AK47”在沒進行處理之前,在10篇文章中出現,那麼它的值為log(300/10)=log30=1.47.

處理之後,“AK47”出現的次數不變,但是總文章數已經變為1100篇,那麼AK47的IDF值為:log(1100/10)=log110=2.04。我們可以看到,經過這樣的處理,軍事的關鍵字都得到了加權,但是經濟,文化的關鍵字的IDF值變化很小。這樣,我們就能夠把軍事的關鍵字同其他的關鍵字區分開來。

我做的實驗中,分出的關鍵字至少80%是軍事類別的,實際的例子就不貼出來了。有的朋友就會問了,那你是怎麼進行文字分類的?難道也是手工分?

嘿嘿,這個當然不是了。分好類的文字來自於搜狗實驗室(http://www.sogou.com/labs/dl/c.html)。上面的類別也只有幾種,如果要做其他類別的樣本,只要用爬蟲抓取某個專業網站或者某一類新聞,然後進行分析出正文就OK了。我們的目標是儘量偷懶,呵呵。

解決了這兩個難題,再回到演算法本身來。首先,提取關鍵字,使得要比較的詞語大大減少(我只提取15%~20%的關鍵字)。一篇1000字的文章詞語也就那麼300~400個,也就是說和一個類別比較50個關鍵字左右就可以判斷出來了,也就是50次的hashmapping操作。然後,有幾個類別就做幾次判斷,所以演算法複雜度是O(m*n)。一般分成十幾個類別已經很細了,整個演算法複雜度不會很高。但是這個實驗我沒能做就申請離職了,傷心啊,我的心血都沒了,如果以後有時間再實驗下吧。