ptgnn:PyTorch图神经网络库

时间:2024-05-06 08:52:42
【文件属性】:

文件名称:ptgnn:PyTorch图神经网络库

文件大小:98KB

文件格式:ZIP

更新时间:2024-05-06 08:52:42

deep-learning pytorch geometric-deep-learning graph-neural-networks gnn

ptgnn:PyTorch GNN库 这是一个包含pyTorch代码的库,用于创建图神经网络(GNN)模型。 该库提供了一些示例实现。 如果您对使用此库感兴趣,请阅读有关它的以及或阅读。 请注意, ptgnn负责定义整个管道,包括数据处理任务,例如数据加载和张紧。 它还为神经网络操作定义了PyTorch nn.Module 。 它们独立于AbstractNeuralModel并且可以用作所有其他PyTorch的nn.Module ,如果愿意的话。 该库主要设计用于快速处理稀疏图。 例如,对于Graph2Class任务上使用默认超参数和结构的V100(以下讨论) ptgnn可以处理约训练期间与约200图表/秒82个图表/秒(209K节点/秒和1,129k边缘/秒)(470K节点/秒和2,527k边/秒)。 已执行的任务 所有任务实现都可以在包中找到。 有关数据和培训步骤的详细说明,请


【文件预览】:
ptgnn-master
----.gitattributes(12B)
----.github()
--------workflows()
----Dockerfile(814B)
----LICENSE(1KB)
----environment.yml(310B)
----setup.py(1KB)
----README.md(6KB)
----.pre-commit-config.yaml(626B)
----SECURITY.md(3KB)
----azure-pipelines.yml(254B)
----docs()
--------architecture.md(12KB)
--------implementations.md(4KB)
--------gnns.md(6KB)
--------azureml.md(2KB)
--------tutorial.md(18KB)
----.editorconfig(151B)
----.amlignore(1KB)
----CODE_OF_CONDUCT.md(444B)
----.gitignore(2KB)
----pyproject.toml(659B)
----ptgnn()
--------baseneuralmodel()
--------__init__.py(0B)
--------py.typed(0B)
--------neuralmodels()
--------implementations()
--------tests()

网友评论