openhgnn.trainerflow.hde_trainer

class hde_trainer(args)[源代码]

HDE trainer flow. Supported Model: HDE Supported Dataset:imdb4hde The trainerflow supports link prediction task. It first calculates HDE for every node in the graph, and uses the HDE as a part of the initial feature for each node. And then it performs standard message passing and link prediction operations. Please notice that for different target node set, the HDE for each node can be different. For more details, please refer to the original paper: http://www.shichuan.org/doc/116.pdf