diff --git a/examples/mxnet/gcn/train.py b/examples/mxnet/gcn/train.py index a20f25f7af94..a9ac59212a87 100644 --- a/examples/mxnet/gcn/train.py +++ b/examples/mxnet/gcn/train.py @@ -1,12 +1,10 @@ """Training GCN model on citation graphs.""" import argparse, time import numpy as np -import networkx as nx import mxnet as mx from mxnet import gluon import dgl -from dgl.data import register_data_args from dgl.data import CoraGraphDataset, CiteseerGraphDataset, PubmedGraphDataset from gcn import GCN @@ -112,7 +110,8 @@ def main(args): if __name__ == '__main__': parser = argparse.ArgumentParser(description='GCN') - register_data_args(parser) + parser.add_argument("--dataset", type=str, default="cora", + help="Dataset name ('cora', 'citeseer', 'pubmed').") parser.add_argument("--dropout", type=float, default=0.5, help="dropout probability") parser.add_argument("--gpu", type=int, default=-1, diff --git a/examples/pytorch/gcn/train.py b/examples/pytorch/gcn/train.py index a2322e7b565d..fda3ceb53244 100644 --- a/examples/pytorch/gcn/train.py +++ b/examples/pytorch/gcn/train.py @@ -1,12 +1,9 @@ import argparse import time import numpy as np -import networkx as nx import torch -import torch.nn as nn import torch.nn.functional as F import dgl -from dgl.data import register_data_args from dgl.data import CoraGraphDataset, CiteseerGraphDataset, PubmedGraphDataset from gcn import GCN @@ -123,7 +120,8 @@ def main(args): if __name__ == '__main__': parser = argparse.ArgumentParser(description='GCN') - register_data_args(parser) + parser.add_argument("--dataset", type=str, default="cora", + help="Dataset name ('cora', 'citeseer', 'pubmed').") parser.add_argument("--dropout", type=float, default=0.5, help="dropout probability") parser.add_argument("--gpu", type=int, default=-1, diff --git a/examples/tensorflow/gcn/train.py b/examples/tensorflow/gcn/train.py index 27bce76525b9..6b53610f710d 100644 --- a/examples/tensorflow/gcn/train.py +++ b/examples/tensorflow/gcn/train.py @@ -1,10 +1,8 @@ import argparse import time import numpy as np -import networkx as nx import tensorflow as tf import dgl -from dgl.data import register_data_args from dgl.data import CoraGraphDataset, CiteseerGraphDataset, PubmedGraphDataset from gcn import GCN @@ -116,7 +114,8 @@ def main(args): if __name__ == '__main__': parser = argparse.ArgumentParser(description='GCN') - register_data_args(parser) + parser.add_argument("--dataset", type=str, default="cora", + help="Dataset name ('cora', 'citeseer', 'pubmed').") parser.add_argument("--dropout", type=float, default=0.5, help="dropout probability") parser.add_argument("--gpu", type=int, default=-1,