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.19k stars 525 forks source link

comirec还有LabelAwareAttention？我看论文是直接argmax #92

Closed chenkkkk closed 1 year ago

chenkkkk commented 1 year ago

https://github.com/shenweichen/DeepMatch/blob/93ca4371386223e2ae0b396b8860f363bd6abf67/deepmatch/models/comirec.py#L173

chenkkkk commented 1 year ago

@liyuan97

shenweichen commented 1 year ago

argmax是label aware的一种特殊形式，我们在代码里做了兼容支持多种模式，可以看下相关的实现@chenkkkk

liyuan97 commented 1 year ago

是吗 我明天确认一下

chenkkkk commented 1 year ago

ok