使用 Pytorch 实现 skip-gram 的 word2vec
转载请注明 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 |