pygip.models.defense.base

Classes

BaseDefense(dataset, attack_node_fraction[, ...])

class pygip.models.defense.base.BaseDefense(dataset, attack_node_fraction, device=None)

Bases: ABC

_abc_impl = <_abc_data object>

_check_dataset_compatibility()

_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.

abstract defend()

Execute the defense mechanism.

supported_api_types = {}

supported_datasets = {}