PROMO

Official Repo of Prompt Tuning for Item Cold-start Recommendation （PROMO）

Datasets

The data pre-processing script is included.

For example, you could download KuaiRand-Pure data from here and run the script to produce the csv format data.

cd data/KuaiRand
sh data_preprocess.sh

To train our model on KuaiRand (with default hyper-parameters):

sh train.sh

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
data		data
models		models
utils		utils
README.md		README.md
__init__.py		__init__.py
dataset.py		dataset.py
main.py		main.py
main_baseline.py		main_baseline.py
main_pretrain.py		main_pretrain.py
model.py		model.py
train.sh		train.sh
train_KuaiRand.ipynb		train_KuaiRand.ipynb