pygip.models.defense.ImperceptibleWM

Module Attributes

class pygip.models.defense.ImperceptibleWM.ImperceptibleWM(dataset, attack_node_fraction=0.3, model_path=None)

Bases: BaseDefense

_abc_impl = <_abc_data object>

_load_model()

Load pre-trained model.

_train_defense_model()

This is an optional method.

_train_surrogate_model()

This is an optional method.

_train_target_model()

This is an optional method.

defend()

Execute the defense mechanism.

supported_api_types = {'pyg'}

class pygip.models.defense.ImperceptibleWM.TriggerGenerator(*args: Any, **kwargs: Any)

Bases: Module

forward(x, edge_index)

pygip.models.defense.ImperceptibleWM.bi_level_optimization(target_model, generator, data, epochs=100, inner_steps=5)

pygip.models.defense.ImperceptibleWM.calculate_metrics(model, data)

pygip.models.defense.ImperceptibleWM.generate_trigger_graph(data, generator, target_model, num_triggers=50)