YuWang-1024 / DSKReG

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DSKReG: Differentiable Sampling on Knowledge Graph for Recommendation with Relational GNN

https://arxiv.org/pdf/2108.11883.pdf

Preprocessing

navigate to src directory: python preprocessing.py -d book

run code

the hyperparameter settings are in the batch_test.py, please uncomment the corresponding settings before runing the code.

python main.py

About


Languages

Language:Python 100.0%