zl程序教程

您现在的位置是:首页 >  后端

当前栏目

使用 Pytorch 实现 skip-gram 的 word2vec

PyTorch 实现 skip Word2Vec 使用
2023-09-27 14:23:28 时间

 

转载请注明 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