machine-learningresearchdeep-learningtensorflowsklearnpytorchdeepwalktransformerconvolutional-neural-networksgcniclrgraph-representationnode2vecgraph-convolutional-networkgraphsagegraph-neural-networksgraph-con
pytorchversion ofhttps://github.com/williamleif/GraphSAGE There are also a handful of new features, including: scripts for preprocessing data attention-based aggregator sparse edge sampler (eg, don't use the dense 2D edgelist) richer, pluggable preprocessing classes ...
pytorch-graphsage/models.py/ Jump to bkjbugfixes Latest commit95bb1bcon Nov 2, 2017History 1contributor 105 lines (82 sloc)3.42 KB RawBlame #!/usr/bin/env python """ models.py """ from__future__importdivision from__future__importprint_function ...
bkj/pytorch-graphsagePublic NotificationsYou must be signed in to change notification settings Fork28 Star138 Files master Sign in to see the full file tree. train.py
这里可以先参考一下GitHub上的GAT代码: class GraphAttentionLayer(nn.Module): """ Simple GAT layer, similar to https://arxiv.org/abs/1710.10903 """ def __init__(self, in_features, out_features, dropout, alpha, concat=True): super(GraphAttentionLayer, self).__init__() self.dropout = ...
This implementation requires Python 3.X. Seerequirements.txtfor a list installed packages and their versions. The main packages are: numpy pytorch scikit-learn Releases No releases published
其实看懂了gcn之后,graphsage基本上也能看懂,graphsage一直强调自己是归纳式的,gcn是直推式的,其实gcn稍微改动一下就可以变成归纳式的,所以其实硬说gcn是直推式的其实对于应用来说帮助不大。 从代码实现上来看, https://github.com/dmlc/dgl/blob/master/python/dgl/nn/pytorch/conv/graphconv.pygithub.com/...
pytorch版本中用的两个数据集都比较小,不是论文里用的数据集。这两个数据集在Kipf 16年经典的GCN论文用到了。节点数量分别约是2700,20000。 cora是一个机器学习论文引用数据集,提供了2708篇论文的引用关系,每篇论文的label是论文所属的领域。label一共七种,包括遗传算法,神经网络,强化学习等7个领域。特征是已经...
[GraphSAGE](https://github.com/williamleif/GraphSAGE)的基本参考PyTorch实现。 这个参考实现对于大图来说没有TensorFlow版本那么快,但是代码更容易阅读,而且在小图的基准上表现得更好(就速度而言)。 该代码也是为了比TensorFlow版本更简单、更可扩展、更容易操作。目前,只有GraphSAGE-mean和GraphSAGE-GCN的监督版本被...
Graph Neural Networks (GNNs) for Node Classification using Graph Attention Networks (GAT) and GraphSAGE, implemented in PyTorch and TensorFlow - MohammadHeydari/GraphSAGE-GAT-Cora-Dataset