Skip to content

Commit

Permalink
AMF
Browse files Browse the repository at this point in the history
  • Loading branch information
hexiangnan committed May 7, 2018
1 parent c896446 commit da003de
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 1 deletion.
15 changes: 14 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,9 @@ APR enhances the pairwise ranking method BPR by performing adversarial training.

This is our official implementation for the paper:

**Adversarial Personalized Ranking for Recommendation** Xiangnan He, Zhankui He, Xiaoyu Du & Tat-Seng Chua, *SIGIR 2018* (Corresponding Author: [Dr. Xiangnan He][http://www.comp.nus.edu.sg/~xiangnan/])
**Adversarial Personalized Ranking for Recommendation** Xiangnan He, Zhankui He, Xiaoyu Du & Tat-Seng Chua, *SIGIR 2018*

(Corresponding Author: [Dr. Xiangnan He][http://www.comp.nus.edu.sg/~xiangnan/])

If you use the codes, please cite our paper . Thanks!

Expand Down Expand Up @@ -40,8 +42,19 @@ python AMF.py --dataset yelp --adv_epoch 1000 --epochs 2000 --eps 0.5 --reg_adv

or `./train.sh yelp` for short.

#### More Details:

Use `python AMF.py -h` to get more argument setting details.

```shell
-h, --help show this help message and exit
--path [PATH] Input data path.
--dataset [DATASET] Choose a dataset.
--verbose VERBOSE Evaluate per X epochs.
--epochs EPOCHS Number of epochs.
......
```

##Dataset

We provide three processed datasets: Yelp(yelp), MovieLens 1 Million (ml-1m) and Pinterest (pinterest-20) in Data/
Expand Down
Binary file modified figure/figure.jpg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit da003de

Please sign in to comment.