gensim訓練詞向量

gensim#encoding=utf-8
from gensim.models import word2vec
sentences=word2vec.Text8Corpus(u'分詞后的爽膚水評論.txt')
model=word2vec.Word2Vec(sentences, size=50)

y2=model.similarity(u"好", u"還行")
print(y2)

for i in model.most_similar(u"滋潤"):
    print i[0],i[1]

txt文件是已經(jīng)分好詞的5W條評論利凑,訓練模型只需一句話:

model=word2vec.Word2Vec(sentences,min_count=5,size=50)

第一個參數(shù)是訓練語料蚊夫,第二個參數(shù)是小于該數(shù)的單詞會被剔除镣陕,默認值為5,
第三個參數(shù)是神經(jīng)網(wǎng)絡的隱藏層單元數(shù),默認為100

model.similarity(u"好", u"還行")#計算兩個詞之間的余弦距離

model.most_similar(u"滋潤")#計算余弦距離最接近“滋潤”的10個詞

運行結果:

0.642981583608
保濕 0.995047152042
溫和 0.985100984573
高 0.978088200092
舒服 0.969187200069
補水 0.967649161816
清爽 0.960570812225
水水 0.958645284176
一般 0.928643763065
一款 0.911774456501
真的 0.90943980217

當然還可以存儲和加載咱們辛辛苦苦訓練好的模型:

model.save('/model/word2vec_model')

new_model=gensim.models.Word2Vec.load('/model/word2vec_model')

也可以獲取每個詞的詞向量

model['computer'] `
?著作權歸作者所有,轉載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末妙蔗,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子疆瑰,更是在濱河造成了極大的恐慌眉反,老刑警劉巖,帶你破解...
    沈念sama閱讀 211,948評論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件穆役,死亡現(xiàn)場離奇詭異寸五,居然都是意外死亡,警方通過查閱死者的電腦和手機孵睬,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,371評論 3 385
  • 文/潘曉璐 我一進店門播歼,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人掰读,你說我怎么就攤上這事秘狞。” “怎么了蹈集?”我有些...
    開封第一講書人閱讀 157,490評論 0 348
  • 文/不壞的土叔 我叫張陵烁试,是天一觀的道長。 經(jīng)常有香客問我拢肆,道長减响,這世上最難降的妖魔是什么靖诗? 我笑而不...
    開封第一講書人閱讀 56,521評論 1 284
  • 正文 為了忘掉前任,我火速辦了婚禮支示,結果婚禮上刊橘,老公的妹妹穿的比我還像新娘。我一直安慰自己颂鸿,他們只是感情好促绵,可當我...
    茶點故事閱讀 65,627評論 6 386
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著嘴纺,像睡著了一般败晴。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上栽渴,一...
    開封第一講書人閱讀 49,842評論 1 290
  • 那天尖坤,我揣著相機與錄音,去河邊找鬼闲擦。 笑死慢味,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的佛致。 我是一名探鬼主播贮缕,決...
    沈念sama閱讀 38,997評論 3 408
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼俺榆!你這毒婦竟也來了感昼?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 37,741評論 0 268
  • 序言:老撾萬榮一對情侶失蹤罐脊,失蹤者是張志新(化名)和其女友劉穎定嗓,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體萍桌,經(jīng)...
    沈念sama閱讀 44,203評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡宵溅,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 36,534評論 2 327
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了上炎。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片恃逻。...
    茶點故事閱讀 38,673評論 1 341
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖藕施,靈堂內(nèi)的尸體忽然破棺而出寇损,到底是詐尸還是另有隱情,我是刑警寧澤裳食,帶...
    沈念sama閱讀 34,339評論 4 330
  • 正文 年R本政府宣布矛市,位于F島的核電站,受9級特大地震影響诲祸,放射性物質(zhì)發(fā)生泄漏浊吏。R本人自食惡果不足惜而昨,卻給世界環(huán)境...
    茶點故事閱讀 39,955評論 3 313
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望找田。 院中可真熱鬧歌憨,春花似錦、人聲如沸午阵。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,770評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽底桂。三九已至,卻和暖如春惧眠,著一層夾襖步出監(jiān)牢的瞬間籽懦,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,000評論 1 266
  • 我被黑心中介騙來泰國打工氛魁, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留暮顺,地道東北人。 一個月前我還...
    沈念sama閱讀 46,394評論 2 360
  • 正文 我出身青樓秀存,卻偏偏與公主長得像捶码,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子或链,可洞房花燭夜當晚...
    茶點故事閱讀 43,562評論 2 349

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