search

westlake-repl / IDvs.MoRec

End-to-end Training for Multimodal Recommendation Systems
Apache License 2.0
121 stars 16 forks source link

Where is the DSSM code #2

Open anthony123 opened 1 year ago

anthony123 commented 1 year ago

Firstly, thanks for your insightful job. I am wondering where is the training code of DSSM?

zyuanat commented 1 year ago

Thanks for the mention and sorry for the late reply. We have updated the DSSM code used in our experiment in bce_vision/main-end2end(DSSM)/'' andbce_text/main-end2end(DSSM)''. The code framework of the DSSM-version is similar to the SASRec-version in other files, the different part is the model.py and encoder.py, and some corresponding settings in run.py, train_xxx.py, and metric.py.

anthony123 commented 1 year ago

Thanks for your reply