02d3197407487646dc5ca6abd889b8fe5fed1aef,examples/pytorch/graphsage/train_sampling.py,,,#,223

Before Change

    val_mask = data.val_mask
    features = th.Tensor(data.features)
    in_feats = features.shape[1]
    labels = th.LongTensor(data.labels)
    n_classes = data.num_labels
    // Construct graph
    g = dgl.graph(data.graph.all_edges())
    g.ndata["features"] = features

After Change

    else:
        device = th.device("cpu")

    if args.dataset == "reddit":
        g, n_classes = load_reddit()
    elif args.dataset == "ogb-product":
        g, n_classes = load_ogb("ogbn-products")
    else:
        raise Exception("unknown dataset")
    g = dgl.as_heterograph(g)
    in_feats = g.ndata["features"].shape[1]
    train_mask = g.ndata["train_mask"]
    val_mask = g.ndata["val_mask"]

Instances