5503ece5e2338f8b2eaa56e17e14b3a2563d7628,demos/link-prediction-graphsage/cora-links-example.py,,test,#Any#Any#Any#,273

Before Change

    G_test = sg.StellarGraph(G_test, node_features="feature")

    // Mapper feeds data from (source, target) sampled subgraphs to GraphSAGE model
    test_mapper = GraphSAGELinkMapper(
        G_test, edge_ids_test, edge_labels_test, batch_size, num_samples, name="test"
    )

    // Evaluate and print metrics
    test_metrics = model.evaluate_generator(test_mapper)

After Change

    G_test = sg.StellarGraph(G_test, node_features="feature")

    // Generator feeds data from (source, target) sampled subgraphs to GraphSAGE model
    test_gen = GraphSAGELinkGenerator(
        G_test,
        batch_size,
        num_samples,
    ).flow(edge_ids_test, edge_labels_test)

    // Evaluate and print metrics
    test_metrics = model.evaluate_generator(test_gen)

Instances