aif360.algorithms.inprocessing.ARTClassifier

class aif360.algorithms.inprocessing.ARTClassifier(art_classifier)

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)

Initialize ARTClassifier.

Parameters:

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

fit(dataset, batch_size=128, nb_epochs=20)

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)

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.