文件名称:DSSM-Lookalike
文件大小:6.35MB
文件格式:ZIP
更新时间:2024-04-09 11:49:34
Python
类似DSSM的管道 论文的实施: 一个基于DSSM的外观类似矢量化召回管道简单实现,包括以下部分: 数据获取:正在使用比赛中的训练数据(100000条)(帮助,uid,标签)。 三种数据负采样方法:<1>随机负采样。<2>根据出现频次负采样。<3> RALM采样方法,参考 DSSM实现,与原论文在损耗计算方式上略有不同。 输出用户嵌入和项目嵌入。 大规模完整的复现了工业界常用的基于DSSM的向量化召回管道。 网络模块建造参考 代码: : 作者:沉伟臣 环境配置 python 3.6.5 张量流== 1.14.0 大熊猫 代码运行示例: python examples/run_dssm_neg_sample.py
【文件预览】:
DSSM-Lookalike-master
----data()
--------train_sample.csv(1.55MB)
--------ml-1m()
----models()
--------dssm.py(3KB)
--------__pycache__()
----examples()
--------run_dssm_movielens.py(8KB)
--------run_dssm_neg_sample.py(3KB)
----utils.py(3KB)
----README.md(1KB)
----.idea()
--------.gitignore(39B)
--------misc.xml(197B)
--------vcs.xml(180B)
--------inspectionProfiles()
--------modules.xml(260B)
--------DSSM.iml(680B)