Skip to content

Commit

Permalink
update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
hyllll committed Nov 19, 2020
1 parent 69e6547 commit abfb31b
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -38,21 +38,18 @@ For clarify, we use MovieLens-100k dataset as a example and treat the BPRMF meth
python data_generator.py
```

3. Next, we run the code to get the results of the teacher model.
3. Next, we run the code to get the results of the teacher model. More details of arguments are available in help message : `python run_pair_mf_train.py --help`

```python
python run_pair_mf_train.py --dataset=ml-100k --prepro=origin
```

More details of arguments are available in help message : `python run_pair_mf_train.py --help`

4. Finally, we train the student model and produce the refined recommendation lists on test set.
4. Finally, we train the student model and produce the refined recommendation lists on test set. More details of arguments are available in help message : `python run_trd.py --help`

```python
python run_trd.py --dataset=ml-100k --prepro=origin --method=bprmf --n_actions=20 --pred_score=0
```

More details of arguments are available in help message : `python run_trd.py --help`

## Acknowledgements

Expand Down

0 comments on commit abfb31b

Please sign in to comment.