anthng / DropEdge

This is a Pytorch implementation of paper: DropEdge: Towards Deep Graph Convolutional Networks on Node Classification

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DropEdge

This is a Pytorch implementation of paper: DropEdge: Towards Deep Graph Convolutional Networks on Node Classification

The data format is this repo. The load_data() function in the utils.py, this function helps organize data in preparation of training.

  # load_data() function in train.py
  
  adj, features, y_train, y_val, y_test, train_mask, val_mask, test_mask = load_data(FLAGS.dataset)
  

About

This is a Pytorch implementation of paper: DropEdge: Towards Deep Graph Convolutional Networks on Node Classification


Languages

Language:Python 71.4%Language:Jupyter Notebook 28.6%