bd298684614646d8f0db5a6597ef88b8d662e25e,tests/defences/test_distillation.py,TestDistillationVectors,test_keras_iris,#TestDistillationVectors#,267

Before Change

                modified_classifier=modified_classifier
            )

        self.assertIn("The input trained classifier do not produce probability outputs.", str(context.exception))

    def test_pytorch_iris(self):
        
        Third test for Pytorch.

After Change

        )

        // Compare the 2 outputs
        preds1 = trained_classifier.predict(
            x=self.x_train_iris,
            batch_size=BATCH_SIZE
        )

        preds2 = modified_classifier.predict(
            x=self.x_train_iris,
            batch_size=BATCH_SIZE
        )

        preds1 = np.argmax(preds1, axis=1)
        preds2 = np.argmax(preds2, axis=1)
        acc = np.sum(preds1 == preds2) / len(preds1)

        self.assertGreater(acc, 0.2)

        ce = cross_entropy(preds1, preds2)

Instances