Skip to content

nosuggest/RDSSM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RDSSM

双塔模型,评论生成系统的一个相似过滤模块。对生成的评论进行相似度计算,保留相似度高的作为candidates

loss graph:

ps:

1.这边采取的是最简单的DNN的双塔,效果已经满足最低的需求了,没有在增加attention,lstm,conv等各种网络

2.这边没有采用原论文的1个正样本+4个负样本的方式,而是1对1的构造二分类,目的是满足query和doc是否相似这种场景的需求

3.这边建议走两个leakyrelu,这边实测单个leakyrelu的情况下loss下降慢,存在欠拟合的情况

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages