grb.attack

grb.attack.base

class grb.attack.base.Attack

Bases: object

Abstract class for graph adversarial attack.

abstract attack(model, adj, features, **kwargs)

Parameters

• model (torch.nn.module) – Model implemented based on torch.nn.module.

• adj (scipy.sparse.csr.csr_matrix) – Adjacency matrix in form of N * N sparse matrix.

• features (torch.FloatTensor) – Features in form of N * D torch float tensor.

• kwargs – Keyword-only arguments.

class grb.attack.base.EarlyStop(patience=1000, epsilon=0.0001)

Bases: object

Strategy to early stop attack process.

class grb.attack.base.InjectionAttack

Bases: Attack

Abstract class for graph injection attack.

abstract attack(**kwargs)

Parameters

kwargs – Keyword-only arguments.

abstract injection(**kwargs)

Parameters

kwargs – Keyword-only arguments.

abstract update_features(**kwargs)

Parameters

kwargs – Keyword-only arguments.

class grb.attack.base.ModificationAttack

Bases: Attack

Abstract class for graph modification attack.

abstract attack(**kwargs)

Parameters

kwargs – Keyword-only arguments.

abstract modification(**kwargs)

Parameters

kwargs – Keyword-only arguments.

grb.attack.fgsm

grb.attack.pgd

grb.attack.rnd

grb.attack.speit

grb.attack.tdgia