Skip to content

RUCAIBox/RecBole-DA

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RecBole-DA

RecBole-DA is a library built upon PyTorch and RecBole for reproducing and developing data augmentation for sequential recommendation.

1)Highlights

  • Easy-to-use API: Our library provides extensive API based on common data augmentation strategies, users can further develop own new models based on our library.
  • Full Coverage of Classic Methods: We provide seven data augmentation methods based on recommender systems in three major categories.

2)Implemented Models

Our library includes algorithms covering three major categories:

  • Heuristic-based Methods: CL4SRec, DuoRec
  • Model-based Methods: MMInfoRec, CauseRec
  • Hybird Methods: CASR, CCL, CoSeRec

3)Requirements

recbole>=1.0.0
pytorch>=1.7.0
python>=3.7.0

4)Quick-Start

With the source code, you can use the provided script for initial usage of our library:

python run_seq.py --dataset='ml-1m' --train_batch_size=256 lmd=0.1 --lmd_sem=0.1 --model='CL4SRec' --contrast='us_x' --sim='dot' --tau=1

If you want to change the models or datasets, just run the script by setting additional command parameters:

python run_seq.py -m [model] -d [dataset]

5)The Team

RecBole-DA is developed and maintained by members from RUCAIBox, the developer is Shuqing Bian (@fancybian).

6) Acknowledgement

CoSeRec and CauseRec are implemented based on CoSeRec and CauseRec. Thanks them for providing efficient implementation.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published