Function test_speit_injection_attack

examples/GRB/test_attack_defense.py:395–427

()

def test_speit_injection_attack():
    graph, dataset, test_mask, device, device_ids = init_dataset()
    model_sur = init_surrogate_model(graph, dataset, test_mask, device, device_ids)
    model_target = init_target_model(graph, dataset, test_mask, device, device_ids)
    print("SPEIT injection attack...")
    inject_modes = ["random", "random-iter", "multi-layer"]
    attacks = [SPEIT(lr=0.02,
                n_epoch=5,
                n_inject_max=10,
                n_edge_max=20,
                feat_lim_min=-1,
                feat_lim_max=1,
                inject_mode=inject_mode,
                early_stop=True,
                early_stop_patience=2,
                early_stop_epsilon=1e-4,
                device=device)
                for inject_mode in inject_modes]
    for attack in attacks:
        graph_attack = attack.attack(model=model_sur,
                                    graph=graph,
                                    adj_norm_func=GCNAdjNorm)
        print(graph_attack)
        test_score_sur = evaluate(model_sur,
                                graph_attack,
                                mask=test_mask,
                                device=device)
        print("Test score after attack for surrogate model: {:.4f}.".format(test_score_sur))
        test_score_target_attack = evaluate(model_target,
                                            graph_attack,
                                            mask=test_mask,
                                            device=device)
        print("Test score after attack for target model: {:.4f}.".format(test_score_target_attack))


def test_tdgia_injection_attack():