gnn python 包_Python gnn包_程序模块 - PyPI - Python中文网

该回购协议包含一个图神经网络模型的tensorflow实现。

安装

要求

gnn框架需要包tensorflow，numpy，scipy。

要安装需求，可以使用以下命令pip install -U -r requirements.txt

安装最新版本的gnn:pip install gnn

有关其他详细信息，请参见Install。

简单用法示例import gnn.GNN as GNN

import gnn.gnn_utils

import Net as n

# Provide your own functions to generate input data

inp, arcnode, nodegraph, labels = set_load()

# Create the state transition function, output function, loss function and metrics

net = n.Net(input_dim, state_dim, output_dim)

# Create the graph neural network model

g = GNN.GNN(net, input_dim, output_dim, state_dim)

#Training

for j in range(0, num_epoch):

g.Train(inp, arcnode, labels, count, nodegraph)

# Validate

print(g.Validate(inp_val, arcnode_val, labels_val, count, nodegraph_val))

许可证

根据3条款BSD许可证发布(请参见license.txt)：Copyright (C) 2004-2019 Matteo Tiezzi

Matteo Tiezzi

Alberto Rossi

欢迎加入QQ群-->： 979659372

推荐PyPI第三方库