轉載請注明 AIQ - 最專業的機器學習大數據社區 http://www.6aiq.com
AIQ 機器學習大數據 知乎專欄 點擊關注
鏈接地址: https://github.com/lonePatient/chinese-word2vec-pytorch
大概 6 次 epochs 之后，可得到一下結果:
| 目標詞 | Top10 | 目標詞 | Top10 |
|---|---|---|---|
| 中國 | 中國 : 1.000 | 男人 | 男人 : 1.000 |
| 中國 | 美國 : 0.651 | 男人 | 女人 : 0.764 |
| 中國 | 日本 : 0.578 | 男人 | 女生 : 0.687 |
| 中國 | 國家 : 0.560 | 男人 | 男生 : 0.670 |
| 中國 | 發展 : 0.550 | 男人 | 喜歡 : 0.625 |
| 中國 | 文化 : 0.529 | 男人 | 戀愛 : 0.601 |
| 中國 | 朝鮮 : 0.512 | 男人 | 歲 : 0.590 |
| 中國 | 經濟 : 0.504 | 男人 | 女 : 0.588 |
| 中國 | 世界 : 0.493 | 男人 | 感覺 : 0.586 |
| 中國 | 社會 : 0.481 | 男人 | 男朋友 : 0.581 |