Function init_surrogate_model

examples/GRB/test_attack_defense.py:55–66

(graph, dataset, test_mask, device, device_ids)

def init_surrogate_model(graph, dataset, test_mask, device, device_ids):
    model_sur = GCN(
        in_feats=graph.num_features,
        hidden_size=16,
        out_feats=graph.num_classes,
        num_layers=2,
        dropout=0.5,
        activation=None
    )
    score_sur = train_model(model_sur, graph, dataset, test_mask, device, device_ids)
    print("Test score before attack for surrogate model: {:.4f}.".format(score_sur))
    return model_sur


def init_target_model(graph, dataset, test_mask, device, device_ids):