使用 Pytorch 实现 skip-gram 的 word2vec
链接地址: 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 |