debrawang / Pytorch-GATNE

根据对论文Representation Learning for Attributed Multiplex Heterogeneous Network的理解和论文源代码用pytorch的toy代码,不对代码模型负责

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Pytorch-GATNE

运行结果:

官方代码:https://github.com/THUDM/GATNE 效果:

Overall ROC-AUC: 0.912732657002574

Overall PR-AUC 0.8709920664369216

Overall F1: 0.8447942506931271

本代码效果:

Overall ROC-AUC: 0.9336145414474002

Overall PR-AUC 0.9088397518884685

Overall F1: 0.8619035458923099

About

根据对论文Representation Learning for Attributed Multiplex Heterogeneous Network的理解和论文源代码用pytorch的toy代码,不对代码模型负责


Languages

Language:Jupyter Notebook 100.0%