├── .gitignore ├── LICENSE ├── README.md ├── datasets └── delicious │ ├── edges.txt │ ├── stats.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── preprocess.py ├── run.py └── srs ├── layers ├── narm.py ├── nextitnet.py ├── seframe.py ├── serec.py ├── srgnn.py ├── ssrm.py └── stamp.py ├── models ├── DGRec.py ├── NARM.py ├── NextItNet.py ├── SERec.py ├── SNARM.py ├── SNextItNet.py ├── SRGNN.py ├── SSRGNN.py ├── SSRM.py ├── SSSRM.py ├── SSTAMP.py └── STAMP.py └── utils ├── Dict.py ├── argparse.py ├── data ├── collate.py ├── load.py ├── preprocess.py └── transform.py ├── prepare_batch.py └── train_runner.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/README.md -------------------------------------------------------------------------------- /datasets/delicious/edges.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/datasets/delicious/edges.txt -------------------------------------------------------------------------------- /datasets/delicious/stats.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/datasets/delicious/stats.txt -------------------------------------------------------------------------------- /datasets/delicious/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/datasets/delicious/test.txt -------------------------------------------------------------------------------- /datasets/delicious/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/datasets/delicious/train.txt -------------------------------------------------------------------------------- /datasets/delicious/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/datasets/delicious/valid.txt -------------------------------------------------------------------------------- /preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/preprocess.py -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/run.py -------------------------------------------------------------------------------- /srs/layers/narm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/layers/narm.py -------------------------------------------------------------------------------- /srs/layers/nextitnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/layers/nextitnet.py -------------------------------------------------------------------------------- /srs/layers/seframe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/layers/seframe.py -------------------------------------------------------------------------------- /srs/layers/serec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/layers/serec.py -------------------------------------------------------------------------------- /srs/layers/srgnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/layers/srgnn.py -------------------------------------------------------------------------------- /srs/layers/ssrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/layers/ssrm.py -------------------------------------------------------------------------------- /srs/layers/stamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/layers/stamp.py -------------------------------------------------------------------------------- /srs/models/DGRec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/models/DGRec.py -------------------------------------------------------------------------------- /srs/models/NARM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/models/NARM.py -------------------------------------------------------------------------------- /srs/models/NextItNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/models/NextItNet.py -------------------------------------------------------------------------------- /srs/models/SERec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/models/SERec.py -------------------------------------------------------------------------------- /srs/models/SNARM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/models/SNARM.py -------------------------------------------------------------------------------- /srs/models/SNextItNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/models/SNextItNet.py -------------------------------------------------------------------------------- /srs/models/SRGNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/models/SRGNN.py -------------------------------------------------------------------------------- /srs/models/SSRGNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/models/SSRGNN.py -------------------------------------------------------------------------------- /srs/models/SSRM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/models/SSRM.py -------------------------------------------------------------------------------- /srs/models/SSSRM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/models/SSSRM.py -------------------------------------------------------------------------------- /srs/models/SSTAMP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/models/SSTAMP.py -------------------------------------------------------------------------------- /srs/models/STAMP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/models/STAMP.py -------------------------------------------------------------------------------- /srs/utils/Dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/utils/Dict.py -------------------------------------------------------------------------------- /srs/utils/argparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/utils/argparse.py -------------------------------------------------------------------------------- /srs/utils/data/collate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/utils/data/collate.py -------------------------------------------------------------------------------- /srs/utils/data/load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/utils/data/load.py -------------------------------------------------------------------------------- /srs/utils/data/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/utils/data/preprocess.py -------------------------------------------------------------------------------- /srs/utils/data/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/utils/data/transform.py -------------------------------------------------------------------------------- /srs/utils/prepare_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/utils/prepare_batch.py -------------------------------------------------------------------------------- /srs/utils/train_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twchen/SEFrame/HEAD/srs/utils/train_runner.py --------------------------------------------------------------------------------