gnn_framework:GNN的C ++框架

时间:2024-04-30 07:40:51
【文件属性】:

文件名称:gnn_framework:GNN的C ++框架

文件大小:42KB

文件格式:ZIP

更新时间:2024-04-30 07:40:51

C++

混合GNN训练框架 安装 要求 C ++ 14 CUDA> = 11.0 联合会 古巴 CUSPARSE 梅蒂斯 从源代码构建 做配置python configure.py 构建全部-j 如何使用 数据集 混合GNN训练框架在COO表示中使用图形数据集。 您可以使用download.py文件安装一些数据集 python download.py 教程 GraphSage @article{hu2020ogb, title={Open Graph Benchmark: Datasets for Machine Learning on Graphs}, author={Weihua Hu, Matthias Fey, Marinka Zitnik, Yuxiao Dong, Hongyu Ren, Bowen Liu, Michele Catasta, Jure Leskove


【文件预览】:
gnn_framework-master
----cuda()
--------aggr_func.cu(6KB)
--------aggr_func.h(2KB)
----model.cc(18KB)
----layers.cc(7KB)
----node.h(1KB)
----hybrid_engine.h(1KB)
----aggregator.h(12KB)
----layers.h(19KB)
----sampler.cc(29KB)
----graph.h(3KB)
----model_params.h(480B)
----utils.cc(2KB)
----sampling_gcn.cc(8KB)
----hybrid_engine.cc(5KB)
----graph.cc(17KB)
----run_exes.py(831B)
----aggregator.cc(61KB)
----model_params.cc(0B)
----utils.h(4KB)
----sampler.h(11KB)
----model.h(3KB)
----Makefile(1KB)
----README.md(707B)
----sampling_gcn.h(0B)
----node.cc(3KB)

网友评论