e1c8ea4f891a3cc98bbcbf8c998f6af895d41fb6,demos/node-classification/graphsage/graphsage-cora-example.py,,,#,254

Before Change

        default=[20, 20],
        help="The number of hidden features at each GraphSAGE layer",
    )
    parser.add_argument(
        "-l",
        "--location",
        type=str,
        default=None,
        help="Location of the CORA dataset (directory)",
    )
    parser.add_argument(
        "-t",
        "--target",
        type=str,
        default="subject",
        help="The target node attribute (categorical)",
    )
    args, cmdline_args = parser.parse_known_args()

    // Load the dataset - this assumes it is the CORA dataset
    // Load graph edgelist
    if args.location is not None:
        graph_loc = os.path.expanduser(args.location)
    else:
        raise ValueError(
            "Please specify the directory containing the dataset using the "-l" flag"
        )

    edgelist = pd.read_csv(
        os.path.join(graph_loc, "cora.cites"),
        sep="\t",
        header=None,

After Change

    args, cmdline_args = parser.parse_known_args()

    // Load the dataset - this assumes it is the CORA dataset and will download it if required:
    dataset = datasets.Cora()
    dataset.download()
    graph_loc = dataset.data_directory
    edgelist = pd.read_csv(
        os.path.join(graph_loc, "cora.cites"),
        sep="\t",
        header=None,

Instances