欢迎使用OpenHGNN¶
OpenHGNN是一个基于 DGL [Deep Graph Library] 和 PyTorch 的开源异质图神经网络工具包, 集成了异质图神经网络的SOTA模型.
关键特性¶
易用: OpenHGNN提供了了易用的接口在给定的模型和数据集上运行实验, 且集成了 optuna 进行超参数优化.
可扩展: 用户可以定义定制化的任务/模型/数据集来对新的场景应用新的模型.
高效: 底层的DGL框架提供了提供了高效的API.
联系我们¶
如果你有任何问题,欢迎提交issue或者邮件联系我们: tyzhao@bupt.edu.cn