-
Notifications
You must be signed in to change notification settings - Fork 26
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Scripts for baseline methods #4
Comments
Thanks for your interest! For ID-based baselines that have been implemented In recbole 1.0.1, like BERT4Rec and those methods that are even not compared In our paper, the scripts are quite similar to SASRec: BERT4Rec:
GRU4Rec:
For baselines with textual features, we slightly modify RecBole's implementations to make sure the textual features are leveraged the same as UniSRec. FDSA:
S^3-Rec pre-training:
S^3-Rec fine-tuning:
For the baseline ZESRec, to the best of our knowledge, the official implementation of ZESRec has not been released, thus we prefer not to upload the corresponding code recently. But I believe that it can be implemented in an easy way. One similar implementation can be regarded as the ablation of UniSRec, by replacing the MoE-enhanced adaptor to one linear module, and removing additional pre-training tasks. For the baselines CCDR and RecGURU, the reported results are based on our own implementations. As the official code of both methods have been released now (Nov. 15), we prefer to reimplement these methods and test their performances thoroughly. Overall, I'll comment again once the implementations of FDSA and S^3-Rec are ready. As FDSA mostly performs the best among baselines, I hope this can help with your reproduction. |
Thank you very much for the comprehensive answer. Looking forward to the implementations! I will close the issue. |
Hi @BoPeng112, we have uploaded the implementations and scripts of FDSA and S^3-Rec, please refer to #4 (comment) for details. Thanks again for your interests!! |
Thanks for the inspiring work! I am trying to reproduce the results of baseline methods. However, the run_baseline.py only works for SASRec. Would you mind sharing the scripts for the other baseline methods?
The text was updated successfully, but these errors were encountered: