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 9f76bc7 commit f7c3a3c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ For clarify, we use MovieLens-100k dataset as a example and treat the BPRMF meth
- ### Data preprocess (data_generator.py)
- Filter the datasets and split the datasets into training set, validation set and test set.
- ### Training the teacher model (run_pair_mf_train.py)
- The teacher model can be well trained by using the historical user-item interaction data. After training, we can get the distilled knowledge (i.e., user and item embeddings as well as the basic recommendation list).
- The teacher model can be well trained by using the historical user-item interaction data. After training, we can get the distilled knowledge (i.e., user and item embeddings as well as the basic recommendation list) from the teacher model.
- ### Training the student model (run_trd.py)
- We treat the distilled knowledge as the input and adopt the Deep Q-Networks (DQN) [1] as the student model. The student model aims to reinforce and refine the basic recommendation list.

Expand Down

0 comments on commit f7c3a3c

Please sign in to comment.