Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
Stevenn9981 committed Jan 31, 2024
1 parent 59ef2aa commit b6d89c7
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions README.md
Expand Up @@ -32,6 +32,12 @@ python run_LightGCN.py --model ppaclg --dataset {dataset} --train

${dataset} can be chosen from ['ml-1M', 'gowalla', 'yelp2018'].

For example:

```
python run_MF.py --model ppacmf --dataset ml-1M --train
```

After training, if you want to use your pre-trained model to conduct inference, use the below script (remove `--train` flag).

```
Expand Down

0 comments on commit b6d89c7

Please sign in to comment.