├── .gitignore ├── README.md ├── data ├── beauty │ ├── datamaps.json │ ├── explanation.json │ ├── negative.txt │ └── sequential.txt ├── sports │ ├── datamaps.json │ ├── explanation.json │ ├── negative.txt │ └── sequential.txt └── toys │ ├── datamaps.json │ ├── explanation.json │ ├── negative.txt │ └── sequential.txt ├── direct.py ├── exp.py ├── model ├── module.py └── templates.py ├── pretrain.py ├── requirements.txt ├── seq_reranker.py └── util ├── bleu.py ├── count_digital_tokens.py ├── rouge.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea/* 2 | *.npy 3 | */__pycache__/* -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/README.md -------------------------------------------------------------------------------- /data/beauty/datamaps.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/data/beauty/datamaps.json -------------------------------------------------------------------------------- /data/beauty/explanation.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/data/beauty/explanation.json -------------------------------------------------------------------------------- /data/beauty/negative.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/data/beauty/negative.txt -------------------------------------------------------------------------------- /data/beauty/sequential.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/data/beauty/sequential.txt -------------------------------------------------------------------------------- /data/sports/datamaps.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/data/sports/datamaps.json -------------------------------------------------------------------------------- /data/sports/explanation.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/data/sports/explanation.json -------------------------------------------------------------------------------- /data/sports/negative.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/data/sports/negative.txt -------------------------------------------------------------------------------- /data/sports/sequential.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/data/sports/sequential.txt -------------------------------------------------------------------------------- /data/toys/datamaps.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/data/toys/datamaps.json -------------------------------------------------------------------------------- /data/toys/explanation.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/data/toys/explanation.json -------------------------------------------------------------------------------- /data/toys/negative.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/data/toys/negative.txt -------------------------------------------------------------------------------- /data/toys/sequential.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/data/toys/sequential.txt -------------------------------------------------------------------------------- /direct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/direct.py -------------------------------------------------------------------------------- /exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/exp.py -------------------------------------------------------------------------------- /model/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/model/module.py -------------------------------------------------------------------------------- /model/templates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/model/templates.py -------------------------------------------------------------------------------- /pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/pretrain.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/requirements.txt -------------------------------------------------------------------------------- /seq_reranker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/seq_reranker.py -------------------------------------------------------------------------------- /util/bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/util/bleu.py -------------------------------------------------------------------------------- /util/count_digital_tokens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/util/count_digital_tokens.py -------------------------------------------------------------------------------- /util/rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/util/rouge.py -------------------------------------------------------------------------------- /util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WangXFng/ELMRec/HEAD/util/utils.py --------------------------------------------------------------------------------