0a78dbe12a90845e1010adcf76f71f04f7386bd1,examples/pytorch/gcn/gcn.py,,main,#Any#,63

Before Change

def main(args):
    // load and preprocess dataset
    if args.dataset == "cora":
        data = load_cora()
    elif args.dataset == "citeseer":
        data = load_citeseer()
    elif args.dataset == "pubmed":
        data = load_pubmed()
    else:
        raise RuntimeError("Error dataset: {}".format(args.dataset))

    // features of each samples
    features = {}
    labels = []
    train_nodes = []
    for n in data.graph.nodes():

After Change

def main(args):
    // load and preprocess dataset
    data = load_data(args)

    // features of each samples
    features = {}
    labels = []

Instances