search

shenweichen / DeepMatch

A deep matching model library for recommendations & advertising. It's easy to train models and to export representation vectors which can be used for ANN search.
https://deepmatch.readthedocs.io/en/latest/
Apache License 2.0
2.22k stars 530 forks source link

DSSM最后一步没有看懂 #33

Closed Ethan199111 closed 2 years ago

Ethan199111 commented 4 years ago

你好，关于DSSM最后一步我有一些疑问

DSSM最后我看是求了user output和item output的余弦距离，然后又进行了sigmoid最后求交叉熵，余弦距离会把得分限制在-1至1之间，这样再进行sigmoid并求交叉熵非常奇怪啊。

我觉得应该求内积后直接使用sigmoid_cross_entropy来得到loss function

我们自己的实践证明上述方法也是有效的，烦请作者说一些这里的思路。

shenweichen commented 4 years ago

这里是仿照DSSM论文进行的实现，使用了cos相似度，代码中也提供了参数可以使用内积，具体的请查阅文档