SPMF

SPMF is a simple and powerful baseline model for collaborative filtering tasks.

Requirements

python>=3.8
torch>=1.8
tqdm

Reproduce

python main.py --model SPMF --dataset yelp2018

python main.py --model SPMF --dataset gowalla

python main.py --model SPMF --dataset amazon-book

python main.py --model PWTC --dataset yelp2018

python main.py --model PWTC --dataset gowalla

python main.py --model PWTC --dataset amazon-book

If you need to view the help, run python main.py -h.

optional arguments:
  -h,  --help           Show this help message and exit
  -m,  --model          Model
  -d,  --dataset        Dataset
  -dp, --dataset_path   Dataset Path
  -bs, --batch_size     Batch Size
  -e,  --epoch          Epoch
  -es, --embedding_size Embedding Size
  -vi, --validation_interval
  -tn, --topN           TopN
  -n                    Iteration Count
  -t                    Temperature Coefficient
  -b                    SPMF Bias
  -p                    PWTC Bias