Skip to content

Latest commit

 

History

History
 
 

deepergcn

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 

DGL Implementation of DeeperGCN

This DGL example implements the GNN model proposed in the paper DeeperGCN: All You Need to Train Deeper GCNs. For the original implementation, see here.

Contributor: xnuohz

Requirements

The codebase is implemented in Python 3.7. For version requirement of packages, see below.

dgl 0.6.0.post1
torch 1.7.0
ogb 1.3.0

The graph datasets used in this example

Open Graph Benchmark(OGB). Dataset summary:

Graph Property Prediction
Dataset #Graphs #Node Feats #Edge Feats Metric
ogbg-molhiv 41,127 9 3 ROC-AUC

Usage

Train a model which follows the original hyperparameters on different datasets.

# ogbg-molhiv
python main.py --gpu 0 --learn-beta

Performance

  • Table 6: Numbers associated with "Table 6" are the ones from table 6 in the paper.
  • Author: Numbers associated with "Author" are the ones we got by running the original code.
  • DGL: Numbers associated with "DGL" are the ones we got by running the DGL example.
Dataset ogbg-molhiv
Results(Table 6) 0.786
Results(Author) 0.781
Results(DGL) 0.778

Speed

Dataset ogbg-molhiv
Results(Author) 11.833
Results(DGL) 8.965