aif360.algorithms.inprocessing.ARTClassifier

class aif360.algorithms.inprocessing.ARTClassifier(art_classifier)[source]

Wraps an instance of an art.classifiers.Classifier to extend Transformer.

Initialize ARTClassifier.

Parameters:art_classifier (art.classifier.Classifier) – A Classifier object from the adversarial-robustness-toolbox.

Methods

fit Train a classifer on the input.
fit_predict Train a model on the input and predict the labels.
fit_transform Train a model on the input and transform the dataset accordingly.
predict Perform prediction for the input.
transform Return a new dataset generated by running this Transformer on the input.
__init__(art_classifier)[source]

Initialize ARTClassifier.

Parameters:art_classifier (art.classifier.Classifier) – A Classifier object from the adversarial-robustness-toolbox.
fit(dataset, batch_size=128, nb_epochs=20)[source]

Train a classifer on the input.

Parameters:
  • dataset (Dataset) – Training dataset.
  • batch_size (int) – Size of batches (passed through to ART).
  • nb_epochs (int) – Number of epochs to use for training (passed through to ART).
Returns:

ARTClassifier – Returns self.
predict(dataset, logits=False)[source]

Perform prediction for the input.

Parameters:
  • dataset (Dataset) – Test dataset.
  • logits (bool, optional) – True is prediction should be done at the logits layer (passed through to ART).
Returns:

Dataset – Dataset with predicted labels in the labels field.