- 授权协议: GPLv3
- 开发语言: Python
- 操作系统: 跨平台
- 软件首页: https://github.com/CrawlScript/TF-GNN
- 软件文档: https://github.com/CrawlScript/TF-GNN
- 官方下载: https://github.com/CrawlScript/TF-GNN
软件介绍
TF-GNN是基于TensorFlow的图神经网络框架,包含了图卷积网络(GCN)和图注意力网络(GAT)等常见的深度图神经网络。框架内部使用了稀疏矩阵等解决了图神经网络的计算效率和规模问题。
项目内容:
图神经网络(如GCN、GAT等)的工程化一直是一个大问题,在数据预处理、网络构建、性能优化等方面都需要细致的考虑。TF-GNN基于TensorFLow,为图神经网络提供了下面的功能:
图神经网络数据集的构建,包括将图神经网络存储格式、图数据的读取、训练测试划分等
基于稀疏矩阵的数据结构大大提升了框架的性能,在CPU上也可以轻松地运行图神经网络
提供了GCN和GAT层,可以直接构建GCN和GAT
数据集:
内置了下面的已经格式化的数据集:
DBLP(论文引用数据集)
M10(论文引用数据集)
示例:
图卷积网络示例(Graph Convolutional Networks):
# coding=utf-8 from gnn.data.dataset import GraphDataset, WhiteSpaceTokenizer from gnn.data.example import load_M10, load_cora, load_dblp from gnn.model.gcn import GCN, GCNTrainer import tensorflow as tf # eager mode must be enabled from tensorflow.contrib.eager.python import tfe import os os.environ["CUDA_VISIBLE_DEVICES"] = "0" tfe.enable_eager_execution() # read graph dataset: data/M10 data/dblp # dataset = GraphDataset("data/dblp", ignore_featureless_node=True) dataset = load_M10("data/M10", ignore_featureless_node=True) adj = dataset.adj_matrix(sparse=True) feature_matrix, feature_masks = dataset.feature_matrix(bag_of_words=True, sparse=True) labels, label_masks = dataset.label_list_or_matrix(one_hot=False) train_node_indices, test_node_indices, train_masks, test_masks = dataset.split_train_and_test(training_rate=0.3) gcn_model = GCN([16, dataset.num_classes()]) gcn_trainer = GCNTrainer(gcn_model) gcn_trainer.train(adj, feature_matrix, labels, train_masks, test_masks, learning_rate=1e-3, l2_coe=1e-3)
Java Web开发从初学到精通
方振宇 / 电子工业 / 2010-6 / 69.00元
《Java Web开发从初学到精通》介绍如何整合Web框架进行J2EE开发,所有实例都基于MyEclipse IDE开发,引领读者快速进入基于JaVa web的J2EE应用领域。《Java Web开发从初学到精通》开始主要介绍Servlet、JSP、JavaBean、EL、JSTL、JDBC等Web开发基础知识,然后学习Struts、Hibernate、Spring、Ajax、JSF等开源框架,并......一起来看看 《Java Web开发从初学到精通》 这本书的介绍吧!