e6839888aa4beb86b59fd79dd9ed951bc1bfb049,demos/link-prediction/graphsage/cora-links-example.py,,train,#Any#Any#Any#Any#Any#Any#Any#,88

Before Change

    // Mapper feeds link data from sampled subgraphs to GraphSAGE model
    // We need to create two mappers: for training and testing of the model
    train_gen = GraphSAGELinkGenerator(
        G_train, batch_size, num_samples, name="train"
    ).flow(edge_ids_train, edge_labels_train, shuffle=True)

    test_gen = GraphSAGELinkGenerator(
        G_test, batch_size, num_samples, name="test"
    ).flow(edge_ids_test, edge_labels_test)

After Change

    // Mapper feeds link data from sampled subgraphs to GraphSAGE model
    // We need to create two mappers: for training and testing of the model
    train_gen = GraphSAGELinkGenerator(G_train, batch_size, num_samples)
    train_flow = train_gen.flow(edge_ids_train, edge_labels_train, shuffle=True)

    test_gen = GraphSAGELinkGenerator(G_test, batch_size, num_samples)
    test_flow = test_gen.flow(edge_ids_test, edge_labels_test)

Instances