46ad98c1b619502c926c657c1a4c4276d2fd1f79,art/defences/adversarial_trainer_unittest.py,TestAdversarialTrainer,test_shared_model_mnist,#TestAdversarialTrainer#,110

Before Change

        // Create and fit classifier
        params = {"epochs": 5, "batch_size": BATCH_SIZE}
        classifier = CNN(im_shape, dataset="mnist")
        classifier.compile({"loss": "categorical_crossentropy", "optimizer": "adam", "metrics": ["accuracy"]})
        classifier.fit(x_train, y_train, **params)

        // Create FGSM attacker

After Change

        adv = FastGradientMethod(classifier)
        x_adv = adv.generate(x_test)
        preds = classifier.predict(x_adv)
        acc = np.sum(np.argmax(preds, axis=1) == np.argmax(y_test, axis=1)) / y_test.shape[0]

        // Perform adversarial training
        adv_trainer = AdversarialTrainer(classifier, adv)
        adv_trainer.fit(x_train, y_train, **params)

Instances