README.md

# PinSage model

NOTE: this version is not using NodeFlow yet.

First, download and extract from https://dgl.ai.s3.us-east-2.amazonaws.com/dataset/ml-1m.tar.gz

One can then run the following to train PinSage on MovieLens-1M:

```bash
python3 train.py --opt Adam --lr 1e-3 --sched none --sgd-switch 25
```

One can also incorporate user and movie features into training:

```bash
python3 train.py --opt Adam --lr 1e-3 --sched none --sgd-switch 25 --use-feature
```

Currently, performance of PinSage on MovieLens-1M has the best mean reciprocal rank of
0.032298±0.048078 on validation (and 0.033695±0.051963 on test set for the same model).
The Implicit Factorization Model from Spotlight has a 0.034572±0.041653 on the test set.