├── README.md ├── baselines ├── D3 │ ├── D3_evaluate.py │ ├── LogitProcesser.py │ └── evaluate2.sh ├── DMPO │ ├── dmpo_trainer.py │ └── utils.py ├── Re-weighting │ └── RW_SFT.py ├── SDPO │ ├── softmax_dpo_trainer.py │ └── utils.py └── Semantic_sampling_rosePO │ └── Semantic_sampling_rosePO.py ├── data └── Goodreads │ ├── test.json │ ├── train.json │ └── valid.json ├── environment.yml ├── eval ├── CDs_and_Vinyl │ ├── embeddings.pt │ ├── genre_dict.json │ ├── id2name.json │ ├── name2genre.json │ └── name2id.json ├── Goodreads │ ├── embeddings.pt │ ├── genre_dict.json │ ├── id2name.json │ ├── name2genre.json │ └── name2id.json ├── MovieLens │ ├── embeddings.pt │ ├── genre_dict.json │ ├── id2name.json │ ├── name2genre.json │ └── name2id.json ├── evaluate.py └── inference.py ├── figs └── method.png ├── shell ├── SFT.sh ├── SPRec.sh └── eval_single_file.sh └── train ├── data_generate.py ├── dpo.py └── sft.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/README.md -------------------------------------------------------------------------------- /baselines/D3/D3_evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/baselines/D3/D3_evaluate.py -------------------------------------------------------------------------------- /baselines/D3/LogitProcesser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/baselines/D3/LogitProcesser.py -------------------------------------------------------------------------------- /baselines/D3/evaluate2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/baselines/D3/evaluate2.sh -------------------------------------------------------------------------------- /baselines/DMPO/dmpo_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/baselines/DMPO/dmpo_trainer.py -------------------------------------------------------------------------------- /baselines/DMPO/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/baselines/DMPO/utils.py -------------------------------------------------------------------------------- /baselines/Re-weighting/RW_SFT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/baselines/Re-weighting/RW_SFT.py -------------------------------------------------------------------------------- /baselines/SDPO/softmax_dpo_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/baselines/SDPO/softmax_dpo_trainer.py -------------------------------------------------------------------------------- /baselines/SDPO/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/baselines/SDPO/utils.py -------------------------------------------------------------------------------- /baselines/Semantic_sampling_rosePO/Semantic_sampling_rosePO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/baselines/Semantic_sampling_rosePO/Semantic_sampling_rosePO.py -------------------------------------------------------------------------------- /data/Goodreads/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/data/Goodreads/test.json -------------------------------------------------------------------------------- /data/Goodreads/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/data/Goodreads/train.json -------------------------------------------------------------------------------- /data/Goodreads/valid.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/data/Goodreads/valid.json -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/environment.yml -------------------------------------------------------------------------------- /eval/CDs_and_Vinyl/embeddings.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/CDs_and_Vinyl/embeddings.pt -------------------------------------------------------------------------------- /eval/CDs_and_Vinyl/genre_dict.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/CDs_and_Vinyl/genre_dict.json -------------------------------------------------------------------------------- /eval/CDs_and_Vinyl/id2name.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/CDs_and_Vinyl/id2name.json -------------------------------------------------------------------------------- /eval/CDs_and_Vinyl/name2genre.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/CDs_and_Vinyl/name2genre.json -------------------------------------------------------------------------------- /eval/CDs_and_Vinyl/name2id.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/CDs_and_Vinyl/name2id.json -------------------------------------------------------------------------------- /eval/Goodreads/embeddings.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/Goodreads/embeddings.pt -------------------------------------------------------------------------------- /eval/Goodreads/genre_dict.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/Goodreads/genre_dict.json -------------------------------------------------------------------------------- /eval/Goodreads/id2name.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/Goodreads/id2name.json -------------------------------------------------------------------------------- /eval/Goodreads/name2genre.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/Goodreads/name2genre.json -------------------------------------------------------------------------------- /eval/Goodreads/name2id.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/Goodreads/name2id.json -------------------------------------------------------------------------------- /eval/MovieLens/embeddings.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/MovieLens/embeddings.pt -------------------------------------------------------------------------------- /eval/MovieLens/genre_dict.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/MovieLens/genre_dict.json -------------------------------------------------------------------------------- /eval/MovieLens/id2name.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/MovieLens/id2name.json -------------------------------------------------------------------------------- /eval/MovieLens/name2genre.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/MovieLens/name2genre.json -------------------------------------------------------------------------------- /eval/MovieLens/name2id.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/MovieLens/name2id.json -------------------------------------------------------------------------------- /eval/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/evaluate.py -------------------------------------------------------------------------------- /eval/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/eval/inference.py -------------------------------------------------------------------------------- /figs/method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/figs/method.png -------------------------------------------------------------------------------- /shell/SFT.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/shell/SFT.sh -------------------------------------------------------------------------------- /shell/SPRec.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/shell/SPRec.sh -------------------------------------------------------------------------------- /shell/eval_single_file.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/shell/eval_single_file.sh -------------------------------------------------------------------------------- /train/data_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/train/data_generate.py -------------------------------------------------------------------------------- /train/dpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/train/dpo.py -------------------------------------------------------------------------------- /train/sft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RegionCh/SPRec/HEAD/train/sft.py --------------------------------------------------------------------------------