├── .gitignore ├── LICENSE ├── README.md ├── dataset ├── .DS_Store ├── beauty │ ├── beauty.test.inter │ └── beauty.train.inter ├── ml-20m │ ├── ml-20m.test.inter │ └── ml-20m.train.inter ├── reddit │ ├── reddit.test.inter │ └── reddit.train.inter └── sports │ ├── sports.test.inter │ └── sports.train.inter ├── figs ├── case.png ├── dataset.png └── model.png ├── recbole ├── __init__.py ├── config │ ├── __init__.py │ └── configurator.py ├── data │ ├── __init__.py │ ├── dataloader │ │ ├── __init__.py │ │ ├── abstract_dataloader.py │ │ ├── general_dataloader.py │ │ ├── knowledge_dataloader.py │ │ └── user_dataloader.py │ ├── dataset │ │ ├── __init__.py │ │ ├── customized_dataset.py │ │ ├── dataset.py │ │ ├── decisiontree_dataset.py │ │ ├── kg_dataset.py │ │ ├── kg_seq_dataset.py │ │ └── sequential_dataset.py │ ├── interaction.py │ └── utils.py ├── dataset_example │ └── ml-100k │ │ ├── ml-100k.inter │ │ ├── ml-100k.item │ │ ├── ml-100k.kg │ │ ├── ml-100k.link │ │ └── ml-100k.user ├── evaluator │ ├── __init__.py │ ├── base_metric.py │ ├── collector.py │ ├── evaluator.py │ ├── metrics.py │ ├── register.py │ └── utils.py ├── model │ ├── __init__.py │ ├── abstract_recommender.py │ ├── context_aware_recommender │ │ ├── __init__.py │ │ ├── afm.py │ │ ├── autoint.py │ │ ├── dcn.py │ │ ├── deepfm.py │ │ ├── dssm.py │ │ ├── ffm.py │ │ ├── fm.py │ │ ├── fnn.py │ │ ├── fwfm.py │ │ ├── lr.py │ │ ├── nfm.py │ │ ├── pnn.py │ │ ├── widedeep.py │ │ └── xdeepfm.py │ ├── exlib_recommender │ │ ├── __init__.py │ │ ├── lightgbm.py │ │ └── xgboost.py │ ├── general_recommender │ │ ├── __init__.py │ │ ├── bpr.py │ │ ├── cdae.py │ │ ├── convncf.py │ │ ├── dgcf.py │ │ ├── dmf.py │ │ ├── ease.py │ │ ├── enmf.py │ │ ├── fism.py │ │ ├── gcmc.py │ │ ├── itemknn.py │ │ ├── lightgcn.py │ │ ├── line.py │ │ ├── macridvae.py │ │ ├── multidae.py │ │ ├── multivae.py │ │ ├── nais.py │ │ ├── neumf.py │ │ ├── ngcf.py │ │ ├── nncf.py │ │ ├── pop.py │ │ ├── ract.py │ │ ├── recvae.py │ │ ├── slimelastic.py │ │ └── spectralcf.py │ ├── init.py │ ├── knowledge_aware_recommender │ │ ├── __init__.py │ │ ├── cfkg.py │ │ ├── cke.py │ │ ├── kgat.py │ │ ├── kgcn.py │ │ ├── kgnnls.py │ │ ├── ktup.py │ │ ├── mkr.py │ │ └── ripplenet.py │ ├── layers.py │ ├── loss.py │ └── sequential_recommender │ │ ├── __init__.py │ │ ├── bert4rec.py │ │ ├── caser.py │ │ ├── dcrec.py │ │ ├── dien.py │ │ ├── din.py │ │ ├── dssrec.py │ │ ├── fdsa.py │ │ ├── fossil.py │ │ ├── fpmc.py │ │ ├── gcsan.py │ │ ├── gru4rec.py │ │ ├── gru4recf.py │ │ ├── gru4reckg.py │ │ ├── hgn.py │ │ ├── hrm.py │ │ ├── insert.py │ │ ├── ksr.py │ │ ├── narm.py │ │ ├── nextitnet.py │ │ ├── npe.py │ │ ├── repeatnet.py │ │ ├── s3rec.py │ │ ├── s3recisp.py │ │ ├── sasrec.py │ │ ├── sasrecf.py │ │ ├── shan.py │ │ ├── srgnn.py │ │ ├── stamp.py │ │ └── transrec.py ├── properties │ ├── dataset │ │ ├── kg_url.yaml │ │ ├── ml-100k.yaml │ │ ├── sample.yaml │ │ └── url.yaml │ ├── model │ │ ├── AFM.yaml │ │ ├── AutoInt.yaml │ │ ├── BERT4Rec.yaml │ │ ├── BPR.yaml │ │ ├── CDAE.yaml │ │ ├── CFKG.yaml │ │ ├── CKE.yaml │ │ ├── Caser.yaml │ │ ├── ConvNCF.yaml │ │ ├── DCN.yaml │ │ ├── DCRec.yaml │ │ ├── DGCF.yaml │ │ ├── DIEN.yaml │ │ ├── DIN.yaml │ │ ├── DMF.yaml │ │ ├── DSSM.yaml │ │ ├── DSSRec.yaml │ │ ├── DeepFM.yaml │ │ ├── EASE.yaml │ │ ├── ENMF.yaml │ │ ├── FDSA.yaml │ │ ├── FFM.yaml │ │ ├── FISM.yaml │ │ ├── FM.yaml │ │ ├── FNN.yaml │ │ ├── FOSSIL.yaml │ │ ├── FPMC.yaml │ │ ├── FwFM.yaml │ │ ├── GCMC.yaml │ │ ├── GCSAN.yaml │ │ ├── GRU4Rec.yaml │ │ ├── GRU4RecF.yaml │ │ ├── GRU4RecKG.yaml │ │ ├── HGN.yaml │ │ ├── HRM.yaml │ │ ├── ItemKNN.yaml │ │ ├── KGAT.yaml │ │ ├── KGCN.yaml │ │ ├── KGNNLS.yaml │ │ ├── KSR.yaml │ │ ├── KTUP.yaml │ │ ├── LINE.yaml │ │ ├── LR.yaml │ │ ├── LightGCN.yaml │ │ ├── MKR.yaml │ │ ├── MacridVAE.yaml │ │ ├── MetaRec.yaml │ │ ├── MetaRecT.yaml │ │ ├── MultiDAE.yaml │ │ ├── MultiVAE.yaml │ │ ├── NAIS.yaml │ │ ├── NARM.yaml │ │ ├── NFM.yaml │ │ ├── NGCF.yaml │ │ ├── NNCF.yaml │ │ ├── NPE.yaml │ │ ├── NeuMF.yaml │ │ ├── NextItNet.yaml │ │ ├── PNN.yaml │ │ ├── Pop.yaml │ │ ├── RaCT.yaml │ │ ├── RecVAE.yaml │ │ ├── RepeatNet.yaml │ │ ├── RippleNet.yaml │ │ ├── S3Rec.yaml │ │ ├── S3RecISP.yaml │ │ ├── SASRec.yaml │ │ ├── SASRecF.yaml │ │ ├── SHAN.yaml │ │ ├── SLIMElastic.yaml │ │ ├── SRGNN.yaml │ │ ├── STAMP.yaml │ │ ├── SpectralCF.yaml │ │ ├── TransRec.yaml │ │ ├── WideDeep.yaml │ │ ├── lightgbm.yaml │ │ ├── xDeepFM.yaml │ │ └── xgboost.yaml │ ├── overall.yaml │ └── quick_start_config │ │ ├── context-aware.yaml │ │ ├── context-aware_ml-100k.yaml │ │ ├── knowledge_base.yaml │ │ ├── sequential.yaml │ │ ├── sequential_DIN.yaml │ │ ├── sequential_DIN_on_ml-100k.yaml │ │ ├── sequential_embedding_model.yaml │ │ └── special_sequential_on_ml-100k.yaml ├── quick_start │ ├── __init__.py │ └── quick_start.py ├── sampler │ ├── __init__.py │ └── sampler.py ├── trainer │ ├── __init__.py │ ├── hyper_tuning.py │ └── trainer.py └── utils │ ├── __init__.py │ ├── argument_list.py │ ├── case_study.py │ ├── enum_type.py │ ├── logger.py │ ├── url.py │ ├── utils.py │ └── wandblogger.py ├── requirements.txt ├── run_DCRec.py └── run_sequential.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/README.md -------------------------------------------------------------------------------- /dataset/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/dataset/.DS_Store -------------------------------------------------------------------------------- /dataset/beauty/beauty.test.inter: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/dataset/beauty/beauty.test.inter -------------------------------------------------------------------------------- /dataset/beauty/beauty.train.inter: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/dataset/beauty/beauty.train.inter -------------------------------------------------------------------------------- /dataset/ml-20m/ml-20m.test.inter: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/dataset/ml-20m/ml-20m.test.inter -------------------------------------------------------------------------------- /dataset/ml-20m/ml-20m.train.inter: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/dataset/ml-20m/ml-20m.train.inter -------------------------------------------------------------------------------- /dataset/reddit/reddit.test.inter: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/dataset/reddit/reddit.test.inter -------------------------------------------------------------------------------- /dataset/reddit/reddit.train.inter: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/dataset/reddit/reddit.train.inter -------------------------------------------------------------------------------- /dataset/sports/sports.test.inter: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/dataset/sports/sports.test.inter -------------------------------------------------------------------------------- /dataset/sports/sports.train.inter: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/dataset/sports/sports.train.inter -------------------------------------------------------------------------------- /figs/case.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/figs/case.png -------------------------------------------------------------------------------- /figs/dataset.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/figs/dataset.png -------------------------------------------------------------------------------- /figs/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/figs/model.png -------------------------------------------------------------------------------- /recbole/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/__init__.py -------------------------------------------------------------------------------- /recbole/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/config/__init__.py -------------------------------------------------------------------------------- /recbole/config/configurator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/config/configurator.py -------------------------------------------------------------------------------- /recbole/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/__init__.py -------------------------------------------------------------------------------- /recbole/data/dataloader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/dataloader/__init__.py -------------------------------------------------------------------------------- /recbole/data/dataloader/abstract_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/dataloader/abstract_dataloader.py -------------------------------------------------------------------------------- /recbole/data/dataloader/general_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/dataloader/general_dataloader.py -------------------------------------------------------------------------------- /recbole/data/dataloader/knowledge_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/dataloader/knowledge_dataloader.py -------------------------------------------------------------------------------- /recbole/data/dataloader/user_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/dataloader/user_dataloader.py -------------------------------------------------------------------------------- /recbole/data/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/dataset/__init__.py -------------------------------------------------------------------------------- /recbole/data/dataset/customized_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/dataset/customized_dataset.py -------------------------------------------------------------------------------- /recbole/data/dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/dataset/dataset.py -------------------------------------------------------------------------------- /recbole/data/dataset/decisiontree_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/dataset/decisiontree_dataset.py -------------------------------------------------------------------------------- /recbole/data/dataset/kg_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/dataset/kg_dataset.py -------------------------------------------------------------------------------- /recbole/data/dataset/kg_seq_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/dataset/kg_seq_dataset.py -------------------------------------------------------------------------------- /recbole/data/dataset/sequential_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/dataset/sequential_dataset.py -------------------------------------------------------------------------------- /recbole/data/interaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/interaction.py -------------------------------------------------------------------------------- /recbole/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/data/utils.py -------------------------------------------------------------------------------- /recbole/dataset_example/ml-100k/ml-100k.inter: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/dataset_example/ml-100k/ml-100k.inter -------------------------------------------------------------------------------- /recbole/dataset_example/ml-100k/ml-100k.item: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/dataset_example/ml-100k/ml-100k.item -------------------------------------------------------------------------------- /recbole/dataset_example/ml-100k/ml-100k.kg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/dataset_example/ml-100k/ml-100k.kg -------------------------------------------------------------------------------- /recbole/dataset_example/ml-100k/ml-100k.link: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/dataset_example/ml-100k/ml-100k.link -------------------------------------------------------------------------------- /recbole/dataset_example/ml-100k/ml-100k.user: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/dataset_example/ml-100k/ml-100k.user -------------------------------------------------------------------------------- /recbole/evaluator/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/evaluator/__init__.py -------------------------------------------------------------------------------- /recbole/evaluator/base_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/evaluator/base_metric.py -------------------------------------------------------------------------------- /recbole/evaluator/collector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/evaluator/collector.py -------------------------------------------------------------------------------- /recbole/evaluator/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/evaluator/evaluator.py -------------------------------------------------------------------------------- /recbole/evaluator/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/evaluator/metrics.py -------------------------------------------------------------------------------- /recbole/evaluator/register.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/evaluator/register.py -------------------------------------------------------------------------------- /recbole/evaluator/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/evaluator/utils.py -------------------------------------------------------------------------------- /recbole/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/__init__.py -------------------------------------------------------------------------------- /recbole/model/abstract_recommender.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/abstract_recommender.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/__init__.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/afm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/afm.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/autoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/autoint.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/dcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/dcn.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/deepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/deepfm.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/dssm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/dssm.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/ffm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/ffm.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/fm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/fm.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/fnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/fnn.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/fwfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/fwfm.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/lr.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/nfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/nfm.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/pnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/pnn.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/widedeep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/widedeep.py -------------------------------------------------------------------------------- /recbole/model/context_aware_recommender/xdeepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/context_aware_recommender/xdeepfm.py -------------------------------------------------------------------------------- /recbole/model/exlib_recommender/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/exlib_recommender/__init__.py -------------------------------------------------------------------------------- /recbole/model/exlib_recommender/lightgbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/exlib_recommender/lightgbm.py -------------------------------------------------------------------------------- /recbole/model/exlib_recommender/xgboost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/exlib_recommender/xgboost.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/__init__.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/bpr.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/cdae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/cdae.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/convncf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/convncf.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/dgcf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/dgcf.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/dmf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/dmf.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/ease.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/ease.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/enmf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/enmf.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/fism.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/fism.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/gcmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/gcmc.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/itemknn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/itemknn.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/lightgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/lightgcn.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/line.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/line.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/macridvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/macridvae.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/multidae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/multidae.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/multivae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/multivae.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/nais.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/nais.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/neumf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/neumf.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/ngcf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/ngcf.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/nncf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/nncf.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/pop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/pop.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/ract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/ract.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/recvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/recvae.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/slimelastic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/slimelastic.py -------------------------------------------------------------------------------- /recbole/model/general_recommender/spectralcf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/general_recommender/spectralcf.py -------------------------------------------------------------------------------- /recbole/model/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/init.py -------------------------------------------------------------------------------- /recbole/model/knowledge_aware_recommender/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/knowledge_aware_recommender/__init__.py -------------------------------------------------------------------------------- /recbole/model/knowledge_aware_recommender/cfkg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/knowledge_aware_recommender/cfkg.py -------------------------------------------------------------------------------- /recbole/model/knowledge_aware_recommender/cke.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/knowledge_aware_recommender/cke.py -------------------------------------------------------------------------------- /recbole/model/knowledge_aware_recommender/kgat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/knowledge_aware_recommender/kgat.py -------------------------------------------------------------------------------- /recbole/model/knowledge_aware_recommender/kgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/knowledge_aware_recommender/kgcn.py -------------------------------------------------------------------------------- /recbole/model/knowledge_aware_recommender/kgnnls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/knowledge_aware_recommender/kgnnls.py -------------------------------------------------------------------------------- /recbole/model/knowledge_aware_recommender/ktup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/knowledge_aware_recommender/ktup.py -------------------------------------------------------------------------------- /recbole/model/knowledge_aware_recommender/mkr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/knowledge_aware_recommender/mkr.py -------------------------------------------------------------------------------- /recbole/model/knowledge_aware_recommender/ripplenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/knowledge_aware_recommender/ripplenet.py -------------------------------------------------------------------------------- /recbole/model/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/layers.py -------------------------------------------------------------------------------- /recbole/model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/loss.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/__init__.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/bert4rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/bert4rec.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/caser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/caser.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/dcrec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/dcrec.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/dien.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/dien.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/din.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/din.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/dssrec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/dssrec.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/fdsa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/fdsa.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/fossil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/fossil.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/fpmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/fpmc.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/gcsan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/gcsan.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/gru4rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/gru4rec.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/gru4recf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/gru4recf.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/gru4reckg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/gru4reckg.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/hgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/hgn.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/hrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/hrm.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/insert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/insert.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/ksr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/ksr.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/narm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/narm.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/nextitnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/nextitnet.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/npe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/npe.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/repeatnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/repeatnet.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/s3rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/s3rec.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/s3recisp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/s3recisp.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/sasrec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/sasrec.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/sasrecf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/sasrecf.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/shan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/shan.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/srgnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/srgnn.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/stamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/stamp.py -------------------------------------------------------------------------------- /recbole/model/sequential_recommender/transrec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/model/sequential_recommender/transrec.py -------------------------------------------------------------------------------- /recbole/properties/dataset/kg_url.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/dataset/kg_url.yaml -------------------------------------------------------------------------------- /recbole/properties/dataset/ml-100k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/dataset/ml-100k.yaml -------------------------------------------------------------------------------- /recbole/properties/dataset/sample.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/dataset/sample.yaml -------------------------------------------------------------------------------- /recbole/properties/dataset/url.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/dataset/url.yaml -------------------------------------------------------------------------------- /recbole/properties/model/AFM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/AFM.yaml -------------------------------------------------------------------------------- /recbole/properties/model/AutoInt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/AutoInt.yaml -------------------------------------------------------------------------------- /recbole/properties/model/BERT4Rec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/BERT4Rec.yaml -------------------------------------------------------------------------------- /recbole/properties/model/BPR.yaml: -------------------------------------------------------------------------------- 1 | embedding_size: 64 -------------------------------------------------------------------------------- /recbole/properties/model/CDAE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/CDAE.yaml -------------------------------------------------------------------------------- /recbole/properties/model/CFKG.yaml: -------------------------------------------------------------------------------- 1 | embedding_size: 64 2 | loss_function: 'inner_product' 3 | margin: 1.0 4 | -------------------------------------------------------------------------------- /recbole/properties/model/CKE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/CKE.yaml -------------------------------------------------------------------------------- /recbole/properties/model/Caser.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/Caser.yaml -------------------------------------------------------------------------------- /recbole/properties/model/ConvNCF.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/ConvNCF.yaml -------------------------------------------------------------------------------- /recbole/properties/model/DCN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/DCN.yaml -------------------------------------------------------------------------------- /recbole/properties/model/DCRec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/DCRec.yaml -------------------------------------------------------------------------------- /recbole/properties/model/DGCF.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/DGCF.yaml -------------------------------------------------------------------------------- /recbole/properties/model/DIEN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/DIEN.yaml -------------------------------------------------------------------------------- /recbole/properties/model/DIN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/DIN.yaml -------------------------------------------------------------------------------- /recbole/properties/model/DMF.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/DMF.yaml -------------------------------------------------------------------------------- /recbole/properties/model/DSSM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/DSSM.yaml -------------------------------------------------------------------------------- /recbole/properties/model/DSSRec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/DSSRec.yaml -------------------------------------------------------------------------------- /recbole/properties/model/DeepFM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/DeepFM.yaml -------------------------------------------------------------------------------- /recbole/properties/model/EASE.yaml: -------------------------------------------------------------------------------- 1 | reg_weight: 250.0 -------------------------------------------------------------------------------- /recbole/properties/model/ENMF.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/ENMF.yaml -------------------------------------------------------------------------------- /recbole/properties/model/FDSA.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/FDSA.yaml -------------------------------------------------------------------------------- /recbole/properties/model/FFM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/FFM.yaml -------------------------------------------------------------------------------- /recbole/properties/model/FISM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/FISM.yaml -------------------------------------------------------------------------------- /recbole/properties/model/FM.yaml: -------------------------------------------------------------------------------- 1 | embedding_size: 10 -------------------------------------------------------------------------------- /recbole/properties/model/FNN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/FNN.yaml -------------------------------------------------------------------------------- /recbole/properties/model/FOSSIL.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/FOSSIL.yaml -------------------------------------------------------------------------------- /recbole/properties/model/FPMC.yaml: -------------------------------------------------------------------------------- 1 | embedding_size: 64 2 | -------------------------------------------------------------------------------- /recbole/properties/model/FwFM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/FwFM.yaml -------------------------------------------------------------------------------- /recbole/properties/model/GCMC.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/GCMC.yaml -------------------------------------------------------------------------------- /recbole/properties/model/GCSAN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/GCSAN.yaml -------------------------------------------------------------------------------- /recbole/properties/model/GRU4Rec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/GRU4Rec.yaml -------------------------------------------------------------------------------- /recbole/properties/model/GRU4RecF.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/GRU4RecF.yaml -------------------------------------------------------------------------------- /recbole/properties/model/GRU4RecKG.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/GRU4RecKG.yaml -------------------------------------------------------------------------------- /recbole/properties/model/HGN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/HGN.yaml -------------------------------------------------------------------------------- /recbole/properties/model/HRM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/HRM.yaml -------------------------------------------------------------------------------- /recbole/properties/model/ItemKNN.yaml: -------------------------------------------------------------------------------- 1 | k: 100 2 | shrink: 0.0 -------------------------------------------------------------------------------- /recbole/properties/model/KGAT.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/KGAT.yaml -------------------------------------------------------------------------------- /recbole/properties/model/KGCN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/KGCN.yaml -------------------------------------------------------------------------------- /recbole/properties/model/KGNNLS.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/KGNNLS.yaml -------------------------------------------------------------------------------- /recbole/properties/model/KSR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/KSR.yaml -------------------------------------------------------------------------------- /recbole/properties/model/KTUP.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/KTUP.yaml -------------------------------------------------------------------------------- /recbole/properties/model/LINE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/LINE.yaml -------------------------------------------------------------------------------- /recbole/properties/model/LR.yaml: -------------------------------------------------------------------------------- 1 | embedding_size: 10 -------------------------------------------------------------------------------- /recbole/properties/model/LightGCN.yaml: -------------------------------------------------------------------------------- 1 | embedding_size: 64 2 | n_layers: 2 3 | reg_weight: 1e-05 -------------------------------------------------------------------------------- /recbole/properties/model/MKR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/MKR.yaml -------------------------------------------------------------------------------- /recbole/properties/model/MacridVAE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/MacridVAE.yaml -------------------------------------------------------------------------------- /recbole/properties/model/MetaRec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/MetaRec.yaml -------------------------------------------------------------------------------- /recbole/properties/model/MetaRecT.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/MetaRecT.yaml -------------------------------------------------------------------------------- /recbole/properties/model/MultiDAE.yaml: -------------------------------------------------------------------------------- 1 | mlp_hidden_size: [600] 2 | latent_dimension: 64 3 | dropout_prob: 0.5 -------------------------------------------------------------------------------- /recbole/properties/model/MultiVAE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/MultiVAE.yaml -------------------------------------------------------------------------------- /recbole/properties/model/NAIS.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/NAIS.yaml -------------------------------------------------------------------------------- /recbole/properties/model/NARM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/NARM.yaml -------------------------------------------------------------------------------- /recbole/properties/model/NFM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/NFM.yaml -------------------------------------------------------------------------------- /recbole/properties/model/NGCF.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/NGCF.yaml -------------------------------------------------------------------------------- /recbole/properties/model/NNCF.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/NNCF.yaml -------------------------------------------------------------------------------- /recbole/properties/model/NPE.yaml: -------------------------------------------------------------------------------- 1 | embedding_size: 64 2 | loss_type: "CE" 3 | dropout_prob: 0.3 -------------------------------------------------------------------------------- /recbole/properties/model/NeuMF.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/NeuMF.yaml -------------------------------------------------------------------------------- /recbole/properties/model/NextItNet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/NextItNet.yaml -------------------------------------------------------------------------------- /recbole/properties/model/PNN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/PNN.yaml -------------------------------------------------------------------------------- /recbole/properties/model/Pop.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /recbole/properties/model/RaCT.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/RaCT.yaml -------------------------------------------------------------------------------- /recbole/properties/model/RecVAE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/RecVAE.yaml -------------------------------------------------------------------------------- /recbole/properties/model/RepeatNet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/RepeatNet.yaml -------------------------------------------------------------------------------- /recbole/properties/model/RippleNet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/RippleNet.yaml -------------------------------------------------------------------------------- /recbole/properties/model/S3Rec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/S3Rec.yaml -------------------------------------------------------------------------------- /recbole/properties/model/S3RecISP.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/S3RecISP.yaml -------------------------------------------------------------------------------- /recbole/properties/model/SASRec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/SASRec.yaml -------------------------------------------------------------------------------- /recbole/properties/model/SASRecF.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/SASRecF.yaml -------------------------------------------------------------------------------- /recbole/properties/model/SHAN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/SHAN.yaml -------------------------------------------------------------------------------- /recbole/properties/model/SLIMElastic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/SLIMElastic.yaml -------------------------------------------------------------------------------- /recbole/properties/model/SRGNN.yaml: -------------------------------------------------------------------------------- 1 | embedding_size: 64 2 | step: 1 3 | loss_type: 'CE' -------------------------------------------------------------------------------- /recbole/properties/model/STAMP.yaml: -------------------------------------------------------------------------------- 1 | embedding_size: 64 2 | loss_type: 'CE' 3 | -------------------------------------------------------------------------------- /recbole/properties/model/SpectralCF.yaml: -------------------------------------------------------------------------------- 1 | embedding_size: 64 2 | n_layers: 4 3 | reg_weight: 1e-03 -------------------------------------------------------------------------------- /recbole/properties/model/TransRec.yaml: -------------------------------------------------------------------------------- 1 | embedding_size: 64 -------------------------------------------------------------------------------- /recbole/properties/model/WideDeep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/WideDeep.yaml -------------------------------------------------------------------------------- /recbole/properties/model/lightgbm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/lightgbm.yaml -------------------------------------------------------------------------------- /recbole/properties/model/xDeepFM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/xDeepFM.yaml -------------------------------------------------------------------------------- /recbole/properties/model/xgboost.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/model/xgboost.yaml -------------------------------------------------------------------------------- /recbole/properties/overall.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/overall.yaml -------------------------------------------------------------------------------- /recbole/properties/quick_start_config/context-aware.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/quick_start_config/context-aware.yaml -------------------------------------------------------------------------------- /recbole/properties/quick_start_config/context-aware_ml-100k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/quick_start_config/context-aware_ml-100k.yaml -------------------------------------------------------------------------------- /recbole/properties/quick_start_config/knowledge_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/quick_start_config/knowledge_base.yaml -------------------------------------------------------------------------------- /recbole/properties/quick_start_config/sequential.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/quick_start_config/sequential.yaml -------------------------------------------------------------------------------- /recbole/properties/quick_start_config/sequential_DIN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/quick_start_config/sequential_DIN.yaml -------------------------------------------------------------------------------- /recbole/properties/quick_start_config/sequential_DIN_on_ml-100k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/quick_start_config/sequential_DIN_on_ml-100k.yaml -------------------------------------------------------------------------------- /recbole/properties/quick_start_config/sequential_embedding_model.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/quick_start_config/sequential_embedding_model.yaml -------------------------------------------------------------------------------- /recbole/properties/quick_start_config/special_sequential_on_ml-100k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/properties/quick_start_config/special_sequential_on_ml-100k.yaml -------------------------------------------------------------------------------- /recbole/quick_start/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/quick_start/__init__.py -------------------------------------------------------------------------------- /recbole/quick_start/quick_start.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/quick_start/quick_start.py -------------------------------------------------------------------------------- /recbole/sampler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/sampler/__init__.py -------------------------------------------------------------------------------- /recbole/sampler/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/sampler/sampler.py -------------------------------------------------------------------------------- /recbole/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/trainer/__init__.py -------------------------------------------------------------------------------- /recbole/trainer/hyper_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/trainer/hyper_tuning.py -------------------------------------------------------------------------------- /recbole/trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/trainer/trainer.py -------------------------------------------------------------------------------- /recbole/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/utils/__init__.py -------------------------------------------------------------------------------- /recbole/utils/argument_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/utils/argument_list.py -------------------------------------------------------------------------------- /recbole/utils/case_study.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/utils/case_study.py -------------------------------------------------------------------------------- /recbole/utils/enum_type.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/utils/enum_type.py -------------------------------------------------------------------------------- /recbole/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/utils/logger.py -------------------------------------------------------------------------------- /recbole/utils/url.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/utils/url.py -------------------------------------------------------------------------------- /recbole/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/utils/utils.py -------------------------------------------------------------------------------- /recbole/utils/wandblogger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/recbole/utils/wandblogger.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_DCRec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/run_DCRec.py -------------------------------------------------------------------------------- /run_sequential.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HKUDS/DCRec/HEAD/run_sequential.py --------------------------------------------------------------------------------