zhjwy9343 / MVP4ModelExample

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DGL Implementation of the CompGCN Paper

This DGL example implements the GNN model proposed in the paper CompositionGCN. The author's codes of implementation is in here

Note: Please replace the paper and author's codes with the paper you implemented.

Example implementor

This example was implemented by zhjwy9343 during his *** work at the AWS Shanghai AI Lab.

Note: Please replace the implementor with your identity and *** with your real status when implement this example.

The graph dataset used in this example

The DGL's built-in CoraGraphDataset. Dataset summary:

  • NumNodes: 2708
  • NumEdges: 10556
  • NumFeats: 1433
  • NumClasses: 7
  • NumTrainingSamples: 140
  • NumValidationSamples: 500
  • NumTestSamples: 1000

How to run example files

In the MVP4ModelExample folder, run

python main.py

If want to use a GPU, run

python main.py --gpu 0

Note:Please replace these commands with your implementation's real commands.

Performance

Note: If your implementation needs to reproduce the SOTA performance, please put these performance here.

About


Languages

Language:Python 100.0%