MKR Model

model base on paper 《Multi-task Learning for KG enhanced Recommendation》

Enviroment

pytorch python3.7

python main.py

we user book rating data to run this model

AUC	ACC
0.710	0.666

Name		Name	Last commit message	Last commit date
Latest commit History 7 Commits
.idea		.idea
MKR-data		MKR-data
__pycache__		__pycache__
.DS_Store		.DS_Store
README.md		README.md
data_process.py		data_process.py
main.py		main.py
mkr_model.py		mkr_model.py
train_model.py		train_model.py
utils.py		utils.py