├── .gitignore ├── LICENSE ├── README.md ├── amazon-c4 └── README.md ├── assets ├── amazon-c4-example.png └── blair.png ├── benchmark_scripts ├── README.md ├── kcore_filtering.py ├── last_out_split.py └── timestamp_split.py ├── blair ├── README.md ├── base.sh ├── large.sh ├── sample_pretraining_data.py ├── simcse │ ├── __init__.py │ ├── models.py │ ├── tool.py │ └── trainers.py ├── test_load_checkpoints.py └── train.py ├── product_search_results ├── README.md ├── bm25.py ├── dataset │ └── process_esci.py ├── eval_search.py └── generate_emb.py └── seq_rec_results ├── README.md ├── config ├── SASRecText.yaml ├── UniSRec.yaml └── overall.yaml ├── data └── dataset.py ├── dataset └── process_amazon_2023.py ├── model ├── sasrectext.py └── unisrec.py ├── run.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/README.md -------------------------------------------------------------------------------- /amazon-c4/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/amazon-c4/README.md -------------------------------------------------------------------------------- /assets/amazon-c4-example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/assets/amazon-c4-example.png -------------------------------------------------------------------------------- /assets/blair.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/assets/blair.png -------------------------------------------------------------------------------- /benchmark_scripts/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/benchmark_scripts/README.md -------------------------------------------------------------------------------- /benchmark_scripts/kcore_filtering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/benchmark_scripts/kcore_filtering.py -------------------------------------------------------------------------------- /benchmark_scripts/last_out_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/benchmark_scripts/last_out_split.py -------------------------------------------------------------------------------- /benchmark_scripts/timestamp_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/benchmark_scripts/timestamp_split.py -------------------------------------------------------------------------------- /blair/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/blair/README.md -------------------------------------------------------------------------------- /blair/base.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/blair/base.sh -------------------------------------------------------------------------------- /blair/large.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/blair/large.sh -------------------------------------------------------------------------------- /blair/sample_pretraining_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/blair/sample_pretraining_data.py -------------------------------------------------------------------------------- /blair/simcse/__init__.py: -------------------------------------------------------------------------------- 1 | from .tool import SimCSE 2 | -------------------------------------------------------------------------------- /blair/simcse/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/blair/simcse/models.py -------------------------------------------------------------------------------- /blair/simcse/tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/blair/simcse/tool.py -------------------------------------------------------------------------------- /blair/simcse/trainers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/blair/simcse/trainers.py -------------------------------------------------------------------------------- /blair/test_load_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/blair/test_load_checkpoints.py -------------------------------------------------------------------------------- /blair/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/blair/train.py -------------------------------------------------------------------------------- /product_search_results/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/product_search_results/README.md -------------------------------------------------------------------------------- /product_search_results/bm25.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/product_search_results/bm25.py -------------------------------------------------------------------------------- /product_search_results/dataset/process_esci.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/product_search_results/dataset/process_esci.py -------------------------------------------------------------------------------- /product_search_results/eval_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/product_search_results/eval_search.py -------------------------------------------------------------------------------- /product_search_results/generate_emb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/product_search_results/generate_emb.py -------------------------------------------------------------------------------- /seq_rec_results/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/seq_rec_results/README.md -------------------------------------------------------------------------------- /seq_rec_results/config/SASRecText.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/seq_rec_results/config/SASRecText.yaml -------------------------------------------------------------------------------- /seq_rec_results/config/UniSRec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/seq_rec_results/config/UniSRec.yaml -------------------------------------------------------------------------------- /seq_rec_results/config/overall.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/seq_rec_results/config/overall.yaml -------------------------------------------------------------------------------- /seq_rec_results/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/seq_rec_results/data/dataset.py -------------------------------------------------------------------------------- /seq_rec_results/dataset/process_amazon_2023.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/seq_rec_results/dataset/process_amazon_2023.py -------------------------------------------------------------------------------- /seq_rec_results/model/sasrectext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/seq_rec_results/model/sasrectext.py -------------------------------------------------------------------------------- /seq_rec_results/model/unisrec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/seq_rec_results/model/unisrec.py -------------------------------------------------------------------------------- /seq_rec_results/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/seq_rec_results/run.py -------------------------------------------------------------------------------- /seq_rec_results/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyp1231/AmazonReviews2023/HEAD/seq_rec_results/utils.py --------------------------------------------------------------------------------