From 37f7d81624042b341f7bf98cbe5af3ca4838169e Mon Sep 17 00:00:00 2001 From: BeiniXie Date: Wed, 28 Dec 2022 10:28:20 +0800 Subject: [PATCH] GRNA --- examples/grna_test.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/examples/grna_test.py b/examples/grna_test.py index 4eec524..d8edb31 100644 --- a/examples/grna_test.py +++ b/examples/grna_test.py @@ -33,10 +33,11 @@ def metattack(data): attack_structure=True, attack_features=False, device='cpu', lambda_=0).to('cpu') # Attack n_perturbations = int(data.edge_index.size(1)/2 * 0.05) + n_perturbations = 1 model.attack(features, adj, labels, idx_train, idx_unlabeled, n_perturbations=n_perturbations, ll_constraint=False) perturbed_adj = model.modified_adj perturbed_data = data.clone() - perturbed_data.edge_index = torch.LongTensor(perturbed_adj.nonzero()) + perturbed_data.edge_index = torch.LongTensor(perturbed_adj.nonzero().T) return perturbed_data @@ -75,6 +76,4 @@ if __name__ == '__main__': data = dataset[0].cpu() dataset[0] = metattack(data).to(device) ptb_acc = test_from_data(trainer, dataset, args) - - - + \ No newline at end of file