说明:类似DSSM的管道
论文的实施:
一个基于DSSM的外观类似矢量化召回管道简单实现,包括以下部分:
数据获取:正在使用比赛中的训练数据(100000条)(帮助,uid,标签)。
三种数据负采样方法:随机负采样。根据出现频次负采样。 RALM采样方法,参考
DSSM实现,与原论文在损耗计算方式上略有不同。
输出用户嵌入和项目嵌入。
大规模完整的复现了工业界常用的基于DSSM的向量化召回管道。
网络模块建造参考
代码: :
作者:沉伟臣
环境配置
python 3.6.5
张量流== 1.1
<weixin_42129113> 上传 | 大小:6mb