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 b0ac8c7 commit f748b5a
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -38,19 +38,15 @@ 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. More details of arguments are available in help message : `python run_pair_mf_train.py --help`

```python
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
```

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
```


## Acknowledgements

We refer to the following repositories to improve our code:
Expand Down

0 comments on commit f748b5a

Please sign in to comment.