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