Rao-Yulong / KDHR

Apache License 2.0
10 stars 5 forks source link

KDHR

This repository is the implementation of KDHR:

Multi-layer information fusion based on graph convolutional network for knowledge-driven herb recommendation

Required packages

torch == 1.5.0+cpu

torch-geometric == 1.4.3 (KDHR and SMGCN) == 1.6.3(other baselines)

numpy == 1.18.1

pandas == 1.0.1

sklearn == 0.22.1