Skip to content

Latest commit

 

History

History
 
 

gcn

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

GCN: Graph Convolutional Networks

Graph Convolutional Network (GCN) is a powerful neural network designed for machine learning on graphs. Based on PGL, we reproduce GCN algorithms and reach the same level of indicators as the paper in citation network benchmarks.

Simple example to build GCN

To build a gcn layer, one can use our pre-defined pgl.nn.GCNConv or just write a gcn layer with message passing interface.

import paddle
import paddle.nn as nn

class CustomGCNConv(nn.Layer):
    def __init__(self, input_size, output_size):
        super(GCNConv, self).__init__()
        self.input_size = input_size
        self.output_size = output_size
        self.linear = nn.Linear(input_size, output_size)
        self.norm = norm
        self.activation = activation

    def forward(self, graph, feature):
        norm = GF.degree_norm(graph)

        feature = self.linear(feature)

        output = graph.send_recv(feature, "sum")

        output = output * norm
        output = nn.functional.relu(output)
        return output


Datasets

The datasets contain three citation networks: CORA, PUBMED, CITESEER. The details for these three datasets can be found in the paper.

Dependencies

  • paddlepaddle==2.0.0
  • pgl==2.1

Performance

We train our models for 200 epochs and report the accuracy on the test dataset.

Dataset Accuracy
Cora ~81%
Pubmed ~79%
Citeseer ~71%

How to run

For examples, use gpu to train gcn on cora dataset.

# Run on GPU
CUDA_VISIBLE_DEVICES=0 python train.py --dataset cora

# Run on CPU 
CUDA_VISIBLE_DEVICES= python train.py --dataset cora

Hyperparameters

  • dataset: The citation dataset "cora", "citeseer", "pubmed".