使用 word2vec 訓(xùn)練wiki中英文語料庫

上學(xué)期讀了有關(guān)word2vec的兩篇paper之后,不是很明白,這學(xué)期重新花時間再讀,并且根據(jù)這兩篇paper進行一個詞向量相關(guān)的實驗,選來選去,發(fā)現(xiàn)網(wǎng)上有大神就wiki中英文語料庫進行訓(xùn)練,鑒于渣渣水平,于是就選擇了訓(xùn)練使用詞向量來訓(xùn)練wiki中英文語料庫。

整個過程參考:“我愛自然語言處理”:www.52nlp.cn


準(zhǔn)備工作

為了訓(xùn)練語料庫,當(dāng)然需要去下載。英文wiki語料庫(11.9G)?中文wiki語料庫(1.2G)

然后就是準(zhǔn)備好編譯環(huán)境,語言選擇的是python,使用了gensim的庫,該庫是由Radim ?eh??ek博士基于google發(fā)布的C語言版本的word2vec編寫的Python庫。如何安裝該庫就不多說了。

處理流程

1、英文wiki訓(xùn)練

首先需要將xml的文件轉(zhuǎn)換成txt文件,主要通過process_wiki.py這個腳本來進行,在參考文考網(wǎng)頁中提出了注意“wiki = WikiCorpus(inp, lemmatize=False, dictionary={})"將lemmatize設(shè)置為False避免使用Pattern來講英文單詞詞干化處理,以免變得很慢,于是就華麗麗的接受。整個過程大概用了5個小時左右,共有差不多400W的articles。

執(zhí)行命令為:python3 process_wiki.py enwiki-latest-pages-articles.xml.bz2 wiki.en.text

生成了wiki.en.text,生成的效果如下:

wiki.en.text效果圖

接下來就是拿wiki.en.text的文本進行詞向量的調(diào)用處理,通過train_word2vec_model.py的腳本對該文件進行處理。

執(zhí)行的命令為:

python3 train_word2vec_model.py wiki.en.text wiki.en.text.model wiki.en.text.vector.?

同樣采取參考blog的方式,保存了vector方便debug。訓(xùn)練了大概8個小時左右,之后在ipython中對已經(jīng)訓(xùn)練好的model進行調(diào)試,調(diào)試如下:

英文wiki結(jié)果圖(一)



英文wiki結(jié)果圖(二)

在這里調(diào)試的時候并沒有遇到參考blog中說到得問題,這里具體原因是什么,還是需要討論。但是可以看出效果其實不怎么好,以后有機會會使用word2vec的C版本進行試驗。當(dāng)然,這里可以使用model的其他函數(shù)進行調(diào)試,具體可以看gensim的文檔進行查看。

2、中文wiki訓(xùn)練

同樣地和英文一樣,使用process_wiki.py 進行腳本處理,處理了10分鐘,執(zhí)行命令為:

python3 process_wiki.py zhwiki-latest-pages-articles.xml.bz2 wiki.cn.text

處理得到的效果如下:


wiki.cn.text效果圖

咦,全是繁體字,那么由于某種原因,wiki中文版本都是繁體字的語料,而且可以看出有一些英文,還有一些其他字符,而且還需要對中文進行分詞,這些都是需要處理的部分。那么我們一步一步的來。


1)繁體轉(zhuǎn)簡體

繁體轉(zhuǎn)成簡體,通過blog得知了一種叫opencc的工具:OpenCC(github地址)

由于是OS X的系統(tǒng),參見安裝說明,但是發(fā)現(xiàn)并沒有安裝brew,于是又去安裝brew,興致沖沖的點進homebrew安裝,結(jié)果發(fā)現(xiàn)主頁無法進去,無奈又開始尋找解決方法,百度之。百度排名第一的英文界面貌似可以解決。

百度brew效果圖

然而點進去,依然無法找到網(wǎng)頁,于是機智的我點了百度快照。


安裝brew代碼

接下來就是一路安裝,完成OpenCC的安裝后。

執(zhí)行命令:opencc -i wiki.cn.text -o wiki.cn.text.jian -c t2s.json

2)分詞

中文不像英文一樣,天生自帶空格,于是就有了中文分詞,在python中中文分詞做的比較好的庫是jieba分詞庫jieba(github地址)

通過代碼庫的實例,于是寫了分詞腳本seperate_words.py.

執(zhí)行命令:python3 separate_words.py wiki.cn.text.jian wiki.cn.text.jian.seq?

得到了分好詞的文件

3)去除多余的其他字符

最后通過python中自帶的re庫,使用正則表達式將其他字符去除,代碼在remove_words.py

執(zhí)行命令:python3 separate_words.py?wiki.cn.text.jian.seq wiki.cn.text.jian.removed


之后就和英文分詞一樣,通過train_word2vec_model.py 進行處理

執(zhí)行命令:python3 train_word2vec_model.py wiki.cn.text.jian.removed wiki.en.text.jian.model wiki.en.text.jian.vector.

最后測試一下運行的效果:


中文wiki結(jié)果圖(一)


中文wiki結(jié)果圖(二)


中文wiki效果圖(三)

代碼:word2vec-for-wiki

----

代碼已經(jīng)進行了review,在python3的情況下,存在bytes和str不兼容的問題已經(jīng)解決了,另外將所有py文件全部放入一個文件中不同函數(shù),在main()函數(shù)中也已經(jīng)進行了各種處理的demo。由于gensim版本變化快,所以在載入model的時候可以參考最新gensim的api。


個人公眾號,分享一下所思所想
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,983評論 6 537
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 98,772評論 3 422
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,947評論 0 381
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經(jīng)常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,201評論 1 315
  • 正文 為了忘掉前任,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 71,960評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 55,350評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,406評論 3 444
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 42,549評論 0 289
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 49,104評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 40,914評論 3 356
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,089評論 1 371
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,647評論 5 362
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 44,340評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,753評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,007評論 1 289
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,834評論 3 395
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 48,106評論 2 375

推薦閱讀更多精彩內(nèi)容