├── .gitignore ├── LICENSE ├── MANIFEST.in ├── README.md ├── README_CN.md ├── assets ├── recstudio_framework.png └── recstudio_logo.png ├── environment.yml ├── example ├── example.ipynb └── sasrec_demo.py ├── nni-experiments ├── config │ ├── bpr.yaml │ └── sasrec.yaml └── search_space │ ├── bpr.yaml │ └── sasrec.yaml ├── recstudio ├── __init__.py ├── ann │ ├── __init__.py │ └── sampler.py ├── data │ ├── __init__.py │ ├── advance_dataset.py │ ├── config │ │ ├── all.yaml │ │ ├── amazon-beauty.yaml │ │ ├── amazon-books.yaml │ │ ├── amazon-electronics.yaml │ │ ├── criteo.yaml │ │ ├── gowalla.yaml │ │ ├── kuairand-pure.yaml │ │ ├── ml-100k.yaml │ │ ├── ml-10m.yaml │ │ ├── ml-1m.yaml │ │ ├── ml-20m.yaml │ │ ├── tmall.yaml │ │ └── yelp.yaml │ └── dataset.py ├── dataset_demo │ ├── __init__.py │ └── ml-100k │ │ ├── ml-100k.inter │ │ ├── ml-100k.item │ │ ├── ml-100k.kg │ │ ├── ml-100k.link │ │ ├── ml-100k.user │ │ └── social.txt ├── eval │ └── __init__.py ├── model │ ├── __init__.py │ ├── ae │ │ ├── config │ │ │ ├── all.yaml │ │ │ ├── multidae.yaml │ │ │ └── multivae.yaml │ │ ├── multidae.py │ │ └── multivae.py │ ├── basemodel │ │ ├── __init__.py │ │ ├── basemodel.yaml │ │ ├── baseranker.py │ │ ├── baseretriever.py │ │ └── recommender.py │ ├── debias │ │ └── __init__.py │ ├── fm │ │ ├── __init__.py │ │ ├── afm.py │ │ ├── afn.py │ │ ├── aoanet.py │ │ ├── autoint.py │ │ ├── ccpm.py │ │ ├── config │ │ │ ├── afm.yaml │ │ │ ├── afn.yaml │ │ │ ├── all.yaml │ │ │ ├── aoanet.yaml │ │ │ ├── autoint.yaml │ │ │ ├── ccpm.yaml │ │ │ ├── dcn.yaml │ │ │ ├── dcnv2.yaml │ │ │ ├── deepcrossing.yaml │ │ │ ├── deepfm.yaml │ │ │ ├── deepim.yaml │ │ │ ├── destine.yaml │ │ │ ├── difm.yaml │ │ │ ├── dlrm.yaml │ │ │ ├── edcn.yaml │ │ │ ├── ffm.yaml │ │ │ ├── fgcnn.yaml │ │ │ ├── fibinet.yaml │ │ │ ├── fignn.yaml │ │ │ ├── finalmlp.yaml │ │ │ ├── flen.yaml │ │ │ ├── fm.yaml │ │ │ ├── fmfm.yaml │ │ │ ├── fwfm.yaml │ │ │ ├── hfm.yaml │ │ │ ├── ifm.yaml │ │ │ ├── interhat.yaml │ │ │ ├── lorentzfm.yaml │ │ │ ├── lr.yaml │ │ │ ├── masknet.yaml │ │ │ ├── nfm.yaml │ │ │ ├── onn.yaml │ │ │ ├── pnn.yaml │ │ │ ├── ppnet.yaml │ │ │ ├── sam.yaml │ │ │ ├── widedeep.yaml │ │ │ └── xdeepfm.yaml │ │ ├── dcn.py │ │ ├── dcnv2.py │ │ ├── deepcrossing.py │ │ ├── deepfm.py │ │ ├── deepim.py │ │ ├── destine.py │ │ ├── difm.py │ │ ├── dlrm.py │ │ ├── edcn.py │ │ ├── ffm.py │ │ ├── fgcnn.py │ │ ├── fibinet.py │ │ ├── fignn.py │ │ ├── finalmlp.py │ │ ├── flen.py │ │ ├── fm.py │ │ ├── fmfm.py │ │ ├── fwfm.py │ │ ├── hfm.py │ │ ├── ifm.py │ │ ├── interhat.py │ │ ├── lorentzfm.py │ │ ├── lr.py │ │ ├── masknet.py │ │ ├── nfm.py │ │ ├── onn.py │ │ ├── pnn.py │ │ ├── ppnet.py │ │ ├── sam.py │ │ ├── widedeep.py │ │ └── xdeepfm.py │ ├── graph │ │ ├── __init__.py │ │ ├── config │ │ │ ├── all.yaml │ │ │ ├── lightgcn.yaml │ │ │ ├── ncl.yaml │ │ │ ├── ngcf.yaml │ │ │ ├── sgl.yaml │ │ │ └── simgcl.yaml │ │ ├── lightgcn.py │ │ ├── ncl.py │ │ ├── ngcf.py │ │ ├── sgl.py │ │ └── simgcl.py │ ├── init.py │ ├── kg │ │ ├── KGLearning.py │ │ ├── __init__.py │ │ ├── cfkg.py │ │ ├── cke.py │ │ ├── config │ │ │ ├── all.yaml │ │ │ ├── cfkg.yaml │ │ │ ├── cke.yaml │ │ │ ├── kgat.yaml │ │ │ ├── kgcn.yaml │ │ │ ├── kgin.yaml │ │ │ ├── kgnnls.yaml │ │ │ ├── ktup.yaml │ │ │ ├── mkr.yaml │ │ │ └── ripplenet.yaml │ │ ├── kgat.py │ │ ├── kgcn.py │ │ ├── kgin.py │ │ ├── kgnnls.py │ │ ├── ktup.py │ │ ├── mkr.py │ │ └── ripplenet.py │ ├── loss_func.py │ ├── mf │ │ ├── __init__.py │ │ ├── bpr.py │ │ ├── cml.py │ │ ├── config │ │ │ ├── all.yaml │ │ │ ├── bpr.yaml │ │ │ ├── cml.yaml │ │ │ ├── dssm.yaml │ │ │ ├── ease.yaml │ │ │ ├── irgan.yaml │ │ │ ├── itemknn.yaml │ │ │ ├── logisticmf.yaml │ │ │ ├── ncf.yaml │ │ │ ├── pmf.yaml │ │ │ ├── slim.yaml │ │ │ └── wrmf.yaml │ │ ├── dssm.py │ │ ├── ease.py │ │ ├── irgan.py │ │ ├── itemknn.py │ │ ├── logisticmf.py │ │ ├── ncf.py │ │ ├── pmf.py │ │ ├── slim.py │ │ └── wrmf.py │ ├── module │ │ ├── __init__.py │ │ ├── ctr.py │ │ ├── data_augmentation.py │ │ ├── functional.py │ │ ├── graphmodule.py │ │ ├── gru.py │ │ └── layers.py │ ├── multitask │ │ ├── __init__.py │ │ ├── aitm.py │ │ ├── config │ │ │ ├── aitm.yaml │ │ │ ├── all.yaml │ │ │ ├── hardshare.yaml │ │ │ ├── mmoe.yaml │ │ │ └── ple.yaml │ │ ├── hardshare.py │ │ ├── mmoe.py │ │ └── ple.py │ ├── ranker.py │ ├── retriever.py │ ├── scorer.py │ └── seq │ │ ├── __init__.py │ │ ├── bert4rec.py │ │ ├── caser.py │ │ ├── cl4srec.py │ │ ├── config │ │ ├── all.yaml │ │ ├── bert4rec.yaml │ │ ├── caser.yaml │ │ ├── cl4srec.yaml │ │ ├── coserec.yaml │ │ ├── din.yaml │ │ ├── fpmc.yaml │ │ ├── gru4rec.yaml │ │ ├── hgn.yaml │ │ ├── iclrec.yaml │ │ ├── narm.yaml │ │ ├── npe.yaml │ │ ├── sasrec.yaml │ │ ├── stamp.yaml │ │ └── transrec.yaml │ │ ├── coserec.py │ │ ├── din.py │ │ ├── fpmc.py │ │ ├── gru4rec.py │ │ ├── hgn.py │ │ ├── iclrec.py │ │ ├── narm.py │ │ ├── npe.py │ │ ├── sasrec.py │ │ ├── stamp.py │ │ └── transrec.py ├── quickstart │ ├── __init__.py │ ├── config_dataset.py │ └── run.py └── utils │ ├── __init__.py │ ├── arguments.py │ ├── callbacks.py │ ├── compress_file.py │ ├── data_parallel.py │ ├── trainer.py │ └── utils.py ├── run.py ├── setup.py └── test ├── test_config_dataset.py ├── test_dataset.py ├── test_ddp.py ├── test_quickrun.py ├── test_retriever.py └── test_training_pipeline.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/README.md -------------------------------------------------------------------------------- /README_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/README_CN.md -------------------------------------------------------------------------------- /assets/recstudio_framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/assets/recstudio_framework.png -------------------------------------------------------------------------------- /assets/recstudio_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/assets/recstudio_logo.png -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/environment.yml -------------------------------------------------------------------------------- /example/example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/example/example.ipynb -------------------------------------------------------------------------------- /example/sasrec_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/example/sasrec_demo.py -------------------------------------------------------------------------------- /nni-experiments/config/bpr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/nni-experiments/config/bpr.yaml -------------------------------------------------------------------------------- /nni-experiments/config/sasrec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/nni-experiments/config/sasrec.yaml -------------------------------------------------------------------------------- /nni-experiments/search_space/bpr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/nni-experiments/search_space/bpr.yaml -------------------------------------------------------------------------------- /nni-experiments/search_space/sasrec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/nni-experiments/search_space/sasrec.yaml -------------------------------------------------------------------------------- /recstudio/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/__init__.py -------------------------------------------------------------------------------- /recstudio/ann/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /recstudio/ann/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/ann/sampler.py -------------------------------------------------------------------------------- /recstudio/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/__init__.py -------------------------------------------------------------------------------- /recstudio/data/advance_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/advance_dataset.py -------------------------------------------------------------------------------- /recstudio/data/config/all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/config/all.yaml -------------------------------------------------------------------------------- /recstudio/data/config/amazon-beauty.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/config/amazon-beauty.yaml -------------------------------------------------------------------------------- /recstudio/data/config/amazon-books.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/config/amazon-books.yaml -------------------------------------------------------------------------------- /recstudio/data/config/amazon-electronics.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/config/amazon-electronics.yaml -------------------------------------------------------------------------------- /recstudio/data/config/criteo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/config/criteo.yaml -------------------------------------------------------------------------------- /recstudio/data/config/gowalla.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/config/gowalla.yaml -------------------------------------------------------------------------------- /recstudio/data/config/kuairand-pure.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/config/kuairand-pure.yaml -------------------------------------------------------------------------------- /recstudio/data/config/ml-100k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/config/ml-100k.yaml -------------------------------------------------------------------------------- /recstudio/data/config/ml-10m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/config/ml-10m.yaml -------------------------------------------------------------------------------- /recstudio/data/config/ml-1m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/config/ml-1m.yaml -------------------------------------------------------------------------------- /recstudio/data/config/ml-20m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/config/ml-20m.yaml -------------------------------------------------------------------------------- /recstudio/data/config/tmall.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/config/tmall.yaml -------------------------------------------------------------------------------- /recstudio/data/config/yelp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/config/yelp.yaml -------------------------------------------------------------------------------- /recstudio/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/data/dataset.py -------------------------------------------------------------------------------- /recstudio/dataset_demo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /recstudio/dataset_demo/ml-100k/ml-100k.inter: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/dataset_demo/ml-100k/ml-100k.inter -------------------------------------------------------------------------------- /recstudio/dataset_demo/ml-100k/ml-100k.item: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/dataset_demo/ml-100k/ml-100k.item -------------------------------------------------------------------------------- /recstudio/dataset_demo/ml-100k/ml-100k.kg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/dataset_demo/ml-100k/ml-100k.kg -------------------------------------------------------------------------------- /recstudio/dataset_demo/ml-100k/ml-100k.link: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/dataset_demo/ml-100k/ml-100k.link -------------------------------------------------------------------------------- /recstudio/dataset_demo/ml-100k/ml-100k.user: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/dataset_demo/ml-100k/ml-100k.user -------------------------------------------------------------------------------- /recstudio/dataset_demo/ml-100k/social.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/dataset_demo/ml-100k/social.txt -------------------------------------------------------------------------------- /recstudio/eval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/eval/__init__.py -------------------------------------------------------------------------------- /recstudio/model/__init__.py: -------------------------------------------------------------------------------- 1 | from . import ae, fm, kg, mf, seq, basemodel -------------------------------------------------------------------------------- /recstudio/model/ae/config/all.yaml: -------------------------------------------------------------------------------- 1 | data: 2 | shuffle: True 3 | -------------------------------------------------------------------------------- /recstudio/model/ae/config/multidae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/ae/config/multidae.yaml -------------------------------------------------------------------------------- /recstudio/model/ae/config/multivae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/ae/config/multivae.yaml -------------------------------------------------------------------------------- /recstudio/model/ae/multidae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/ae/multidae.py -------------------------------------------------------------------------------- /recstudio/model/ae/multivae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/ae/multivae.py -------------------------------------------------------------------------------- /recstudio/model/basemodel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/basemodel/__init__.py -------------------------------------------------------------------------------- /recstudio/model/basemodel/basemodel.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/basemodel/basemodel.yaml -------------------------------------------------------------------------------- /recstudio/model/basemodel/baseranker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/basemodel/baseranker.py -------------------------------------------------------------------------------- /recstudio/model/basemodel/baseretriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/basemodel/baseretriever.py -------------------------------------------------------------------------------- /recstudio/model/basemodel/recommender.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/basemodel/recommender.py -------------------------------------------------------------------------------- /recstudio/model/debias/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /recstudio/model/fm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/__init__.py -------------------------------------------------------------------------------- /recstudio/model/fm/afm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/afm.py -------------------------------------------------------------------------------- /recstudio/model/fm/afn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/afn.py -------------------------------------------------------------------------------- /recstudio/model/fm/aoanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/aoanet.py -------------------------------------------------------------------------------- /recstudio/model/fm/autoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/autoint.py -------------------------------------------------------------------------------- /recstudio/model/fm/ccpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/ccpm.py -------------------------------------------------------------------------------- /recstudio/model/fm/config/afm.yaml: -------------------------------------------------------------------------------- 1 | model: 2 | attention_dim: 4 3 | dropout: 0.5 -------------------------------------------------------------------------------- /recstudio/model/fm/config/afn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/afn.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/all.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/aoanet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/aoanet.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/autoint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/autoint.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/ccpm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/ccpm.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/dcn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/dcn.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/dcnv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/dcnv2.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/deepcrossing.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/deepcrossing.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/deepfm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/deepfm.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/deepim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/deepim.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/destine.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/destine.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/difm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/difm.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/dlrm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/dlrm.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/edcn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/edcn.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/ffm.yaml: -------------------------------------------------------------------------------- 1 | model: 2 | embed_dim: 10 -------------------------------------------------------------------------------- /recstudio/model/fm/config/fgcnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/fgcnn.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/fibinet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/fibinet.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/fignn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/fignn.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/finalmlp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/finalmlp.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/flen.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/flen.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/fm.yaml: -------------------------------------------------------------------------------- 1 | model: 2 | embed_dim: 10 3 | -------------------------------------------------------------------------------- /recstudio/model/fm/config/fmfm.yaml: -------------------------------------------------------------------------------- 1 | model: 2 | embed_dim: 10 -------------------------------------------------------------------------------- /recstudio/model/fm/config/fwfm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/fwfm.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/hfm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/hfm.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/ifm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/ifm.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/interhat.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/interhat.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/lorentzfm.yaml: -------------------------------------------------------------------------------- 1 | model: 2 | embed_dim: 10 -------------------------------------------------------------------------------- /recstudio/model/fm/config/lr.yaml: -------------------------------------------------------------------------------- 1 | model: 2 | embed_dim: 1 -------------------------------------------------------------------------------- /recstudio/model/fm/config/masknet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/masknet.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/nfm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/nfm.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/onn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/onn.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/pnn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/pnn.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/ppnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/ppnet.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/sam.yaml: -------------------------------------------------------------------------------- 1 | model: 2 | dropout: 0 3 | interaction_type: sam2e -------------------------------------------------------------------------------- /recstudio/model/fm/config/widedeep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/widedeep.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/config/xdeepfm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/config/xdeepfm.yaml -------------------------------------------------------------------------------- /recstudio/model/fm/dcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/dcn.py -------------------------------------------------------------------------------- /recstudio/model/fm/dcnv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/dcnv2.py -------------------------------------------------------------------------------- /recstudio/model/fm/deepcrossing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/deepcrossing.py -------------------------------------------------------------------------------- /recstudio/model/fm/deepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/deepfm.py -------------------------------------------------------------------------------- /recstudio/model/fm/deepim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/deepim.py -------------------------------------------------------------------------------- /recstudio/model/fm/destine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/destine.py -------------------------------------------------------------------------------- /recstudio/model/fm/difm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/difm.py -------------------------------------------------------------------------------- /recstudio/model/fm/dlrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/dlrm.py -------------------------------------------------------------------------------- /recstudio/model/fm/edcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/edcn.py -------------------------------------------------------------------------------- /recstudio/model/fm/ffm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/ffm.py -------------------------------------------------------------------------------- /recstudio/model/fm/fgcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/fgcnn.py -------------------------------------------------------------------------------- /recstudio/model/fm/fibinet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/fibinet.py -------------------------------------------------------------------------------- /recstudio/model/fm/fignn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/fignn.py -------------------------------------------------------------------------------- /recstudio/model/fm/finalmlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/finalmlp.py -------------------------------------------------------------------------------- /recstudio/model/fm/flen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/flen.py -------------------------------------------------------------------------------- /recstudio/model/fm/fm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/fm.py -------------------------------------------------------------------------------- /recstudio/model/fm/fmfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/fmfm.py -------------------------------------------------------------------------------- /recstudio/model/fm/fwfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/fwfm.py -------------------------------------------------------------------------------- /recstudio/model/fm/hfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/hfm.py -------------------------------------------------------------------------------- /recstudio/model/fm/ifm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/ifm.py -------------------------------------------------------------------------------- /recstudio/model/fm/interhat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/interhat.py -------------------------------------------------------------------------------- /recstudio/model/fm/lorentzfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/lorentzfm.py -------------------------------------------------------------------------------- /recstudio/model/fm/lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/lr.py -------------------------------------------------------------------------------- /recstudio/model/fm/masknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/masknet.py -------------------------------------------------------------------------------- /recstudio/model/fm/nfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/nfm.py -------------------------------------------------------------------------------- /recstudio/model/fm/onn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/onn.py -------------------------------------------------------------------------------- /recstudio/model/fm/pnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/pnn.py -------------------------------------------------------------------------------- /recstudio/model/fm/ppnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/ppnet.py -------------------------------------------------------------------------------- /recstudio/model/fm/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/sam.py -------------------------------------------------------------------------------- /recstudio/model/fm/widedeep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/widedeep.py -------------------------------------------------------------------------------- /recstudio/model/fm/xdeepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/fm/xdeepfm.py -------------------------------------------------------------------------------- /recstudio/model/graph/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/graph/__init__.py -------------------------------------------------------------------------------- /recstudio/model/graph/config/all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/graph/config/all.yaml -------------------------------------------------------------------------------- /recstudio/model/graph/config/lightgcn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/graph/config/lightgcn.yaml -------------------------------------------------------------------------------- /recstudio/model/graph/config/ncl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/graph/config/ncl.yaml -------------------------------------------------------------------------------- /recstudio/model/graph/config/ngcf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/graph/config/ngcf.yaml -------------------------------------------------------------------------------- /recstudio/model/graph/config/sgl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/graph/config/sgl.yaml -------------------------------------------------------------------------------- /recstudio/model/graph/config/simgcl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/graph/config/simgcl.yaml -------------------------------------------------------------------------------- /recstudio/model/graph/lightgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/graph/lightgcn.py -------------------------------------------------------------------------------- /recstudio/model/graph/ncl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/graph/ncl.py -------------------------------------------------------------------------------- /recstudio/model/graph/ngcf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/graph/ngcf.py -------------------------------------------------------------------------------- /recstudio/model/graph/sgl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/graph/sgl.py -------------------------------------------------------------------------------- /recstudio/model/graph/simgcl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/graph/simgcl.py -------------------------------------------------------------------------------- /recstudio/model/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/init.py -------------------------------------------------------------------------------- /recstudio/model/kg/KGLearning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/KGLearning.py -------------------------------------------------------------------------------- /recstudio/model/kg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /recstudio/model/kg/cfkg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/cfkg.py -------------------------------------------------------------------------------- /recstudio/model/kg/cke.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/cke.py -------------------------------------------------------------------------------- /recstudio/model/kg/config/all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/config/all.yaml -------------------------------------------------------------------------------- /recstudio/model/kg/config/cfkg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/config/cfkg.yaml -------------------------------------------------------------------------------- /recstudio/model/kg/config/cke.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/config/cke.yaml -------------------------------------------------------------------------------- /recstudio/model/kg/config/kgat.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/config/kgat.yaml -------------------------------------------------------------------------------- /recstudio/model/kg/config/kgcn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/config/kgcn.yaml -------------------------------------------------------------------------------- /recstudio/model/kg/config/kgin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/config/kgin.yaml -------------------------------------------------------------------------------- /recstudio/model/kg/config/kgnnls.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/config/kgnnls.yaml -------------------------------------------------------------------------------- /recstudio/model/kg/config/ktup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/config/ktup.yaml -------------------------------------------------------------------------------- /recstudio/model/kg/config/mkr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/config/mkr.yaml -------------------------------------------------------------------------------- /recstudio/model/kg/config/ripplenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/config/ripplenet.yaml -------------------------------------------------------------------------------- /recstudio/model/kg/kgat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/kgat.py -------------------------------------------------------------------------------- /recstudio/model/kg/kgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/kgcn.py -------------------------------------------------------------------------------- /recstudio/model/kg/kgin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/kgin.py -------------------------------------------------------------------------------- /recstudio/model/kg/kgnnls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/kgnnls.py -------------------------------------------------------------------------------- /recstudio/model/kg/ktup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/ktup.py -------------------------------------------------------------------------------- /recstudio/model/kg/mkr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/mkr.py -------------------------------------------------------------------------------- /recstudio/model/kg/ripplenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/kg/ripplenet.py -------------------------------------------------------------------------------- /recstudio/model/loss_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/loss_func.py -------------------------------------------------------------------------------- /recstudio/model/mf/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/__init__.py -------------------------------------------------------------------------------- /recstudio/model/mf/bpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/bpr.py -------------------------------------------------------------------------------- /recstudio/model/mf/cml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/cml.py -------------------------------------------------------------------------------- /recstudio/model/mf/config/all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/config/all.yaml -------------------------------------------------------------------------------- /recstudio/model/mf/config/bpr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/config/bpr.yaml -------------------------------------------------------------------------------- /recstudio/model/mf/config/cml.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/config/cml.yaml -------------------------------------------------------------------------------- /recstudio/model/mf/config/dssm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/config/dssm.yaml -------------------------------------------------------------------------------- /recstudio/model/mf/config/ease.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/config/ease.yaml -------------------------------------------------------------------------------- /recstudio/model/mf/config/irgan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/config/irgan.yaml -------------------------------------------------------------------------------- /recstudio/model/mf/config/itemknn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/config/itemknn.yaml -------------------------------------------------------------------------------- /recstudio/model/mf/config/logisticmf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/config/logisticmf.yaml -------------------------------------------------------------------------------- /recstudio/model/mf/config/ncf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/config/ncf.yaml -------------------------------------------------------------------------------- /recstudio/model/mf/config/pmf.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /recstudio/model/mf/config/slim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/config/slim.yaml -------------------------------------------------------------------------------- /recstudio/model/mf/config/wrmf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/config/wrmf.yaml -------------------------------------------------------------------------------- /recstudio/model/mf/dssm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/dssm.py -------------------------------------------------------------------------------- /recstudio/model/mf/ease.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/ease.py -------------------------------------------------------------------------------- /recstudio/model/mf/irgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/irgan.py -------------------------------------------------------------------------------- /recstudio/model/mf/itemknn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/itemknn.py -------------------------------------------------------------------------------- /recstudio/model/mf/logisticmf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/logisticmf.py -------------------------------------------------------------------------------- /recstudio/model/mf/ncf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/ncf.py -------------------------------------------------------------------------------- /recstudio/model/mf/pmf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/pmf.py -------------------------------------------------------------------------------- /recstudio/model/mf/slim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/slim.py -------------------------------------------------------------------------------- /recstudio/model/mf/wrmf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/mf/wrmf.py -------------------------------------------------------------------------------- /recstudio/model/module/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/module/__init__.py -------------------------------------------------------------------------------- /recstudio/model/module/ctr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/module/ctr.py -------------------------------------------------------------------------------- /recstudio/model/module/data_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/module/data_augmentation.py -------------------------------------------------------------------------------- /recstudio/model/module/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/module/functional.py -------------------------------------------------------------------------------- /recstudio/model/module/graphmodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/module/graphmodule.py -------------------------------------------------------------------------------- /recstudio/model/module/gru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/module/gru.py -------------------------------------------------------------------------------- /recstudio/model/module/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/module/layers.py -------------------------------------------------------------------------------- /recstudio/model/multitask/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/multitask/__init__.py -------------------------------------------------------------------------------- /recstudio/model/multitask/aitm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/multitask/aitm.py -------------------------------------------------------------------------------- /recstudio/model/multitask/config/aitm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/multitask/config/aitm.yaml -------------------------------------------------------------------------------- /recstudio/model/multitask/config/all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/multitask/config/all.yaml -------------------------------------------------------------------------------- /recstudio/model/multitask/config/hardshare.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/multitask/config/hardshare.yaml -------------------------------------------------------------------------------- /recstudio/model/multitask/config/mmoe.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/multitask/config/mmoe.yaml -------------------------------------------------------------------------------- /recstudio/model/multitask/config/ple.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/multitask/config/ple.yaml -------------------------------------------------------------------------------- /recstudio/model/multitask/hardshare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/multitask/hardshare.py -------------------------------------------------------------------------------- /recstudio/model/multitask/mmoe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/multitask/mmoe.py -------------------------------------------------------------------------------- /recstudio/model/multitask/ple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/multitask/ple.py -------------------------------------------------------------------------------- /recstudio/model/ranker.py: -------------------------------------------------------------------------------- 1 | from recstudio.model.fm import * -------------------------------------------------------------------------------- /recstudio/model/retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/retriever.py -------------------------------------------------------------------------------- /recstudio/model/scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/scorer.py -------------------------------------------------------------------------------- /recstudio/model/seq/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/__init__.py -------------------------------------------------------------------------------- /recstudio/model/seq/bert4rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/bert4rec.py -------------------------------------------------------------------------------- /recstudio/model/seq/caser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/caser.py -------------------------------------------------------------------------------- /recstudio/model/seq/cl4srec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/cl4srec.py -------------------------------------------------------------------------------- /recstudio/model/seq/config/all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/config/all.yaml -------------------------------------------------------------------------------- /recstudio/model/seq/config/bert4rec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/config/bert4rec.yaml -------------------------------------------------------------------------------- /recstudio/model/seq/config/caser.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/config/caser.yaml -------------------------------------------------------------------------------- /recstudio/model/seq/config/cl4srec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/config/cl4srec.yaml -------------------------------------------------------------------------------- /recstudio/model/seq/config/coserec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/config/coserec.yaml -------------------------------------------------------------------------------- /recstudio/model/seq/config/din.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/config/din.yaml -------------------------------------------------------------------------------- /recstudio/model/seq/config/fpmc.yaml: -------------------------------------------------------------------------------- 1 | train: 2 | negative_count: 1 -------------------------------------------------------------------------------- /recstudio/model/seq/config/gru4rec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/config/gru4rec.yaml -------------------------------------------------------------------------------- /recstudio/model/seq/config/hgn.yaml: -------------------------------------------------------------------------------- 1 | model: 2 | pooling_type: mean 3 | 4 | train: 5 | negative_count: 1 6 | -------------------------------------------------------------------------------- /recstudio/model/seq/config/iclrec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/config/iclrec.yaml -------------------------------------------------------------------------------- /recstudio/model/seq/config/narm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/config/narm.yaml -------------------------------------------------------------------------------- /recstudio/model/seq/config/npe.yaml: -------------------------------------------------------------------------------- 1 | model: 2 | dropout_rate: 0.3 3 | 4 | train: 5 | negative_count: 1 6 | -------------------------------------------------------------------------------- /recstudio/model/seq/config/sasrec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/config/sasrec.yaml -------------------------------------------------------------------------------- /recstudio/model/seq/config/stamp.yaml: -------------------------------------------------------------------------------- 1 | model: 2 | embed_dim: 64 3 | -------------------------------------------------------------------------------- /recstudio/model/seq/config/transrec.yaml: -------------------------------------------------------------------------------- 1 | trian: 2 | negative_count: 1 -------------------------------------------------------------------------------- /recstudio/model/seq/coserec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/coserec.py -------------------------------------------------------------------------------- /recstudio/model/seq/din.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/din.py -------------------------------------------------------------------------------- /recstudio/model/seq/fpmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/fpmc.py -------------------------------------------------------------------------------- /recstudio/model/seq/gru4rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/gru4rec.py -------------------------------------------------------------------------------- /recstudio/model/seq/hgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/hgn.py -------------------------------------------------------------------------------- /recstudio/model/seq/iclrec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/iclrec.py -------------------------------------------------------------------------------- /recstudio/model/seq/narm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/narm.py -------------------------------------------------------------------------------- /recstudio/model/seq/npe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/npe.py -------------------------------------------------------------------------------- /recstudio/model/seq/sasrec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/sasrec.py -------------------------------------------------------------------------------- /recstudio/model/seq/stamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/stamp.py -------------------------------------------------------------------------------- /recstudio/model/seq/transrec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/model/seq/transrec.py -------------------------------------------------------------------------------- /recstudio/quickstart/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/quickstart/__init__.py -------------------------------------------------------------------------------- /recstudio/quickstart/config_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/quickstart/config_dataset.py -------------------------------------------------------------------------------- /recstudio/quickstart/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/quickstart/run.py -------------------------------------------------------------------------------- /recstudio/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/utils/__init__.py -------------------------------------------------------------------------------- /recstudio/utils/arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/utils/arguments.py -------------------------------------------------------------------------------- /recstudio/utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/utils/callbacks.py -------------------------------------------------------------------------------- /recstudio/utils/compress_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/utils/compress_file.py -------------------------------------------------------------------------------- /recstudio/utils/data_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/utils/data_parallel.py -------------------------------------------------------------------------------- /recstudio/utils/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/utils/trainer.py -------------------------------------------------------------------------------- /recstudio/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/recstudio/utils/utils.py -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/run.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/setup.py -------------------------------------------------------------------------------- /test/test_config_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/test/test_config_dataset.py -------------------------------------------------------------------------------- /test/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/test/test_dataset.py -------------------------------------------------------------------------------- /test/test_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/test/test_ddp.py -------------------------------------------------------------------------------- /test/test_quickrun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/test/test_quickrun.py -------------------------------------------------------------------------------- /test/test_retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/test/test_retriever.py -------------------------------------------------------------------------------- /test/test_training_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/USTCLLM/RecStudio/HEAD/test/test_training_pipeline.py --------------------------------------------------------------------------------