c7585e649aca88cf9e21eb754469630748bb1cfc,tests/attacks/test_knockoff_nets.py,TestKnockoffNets,test_pytorch_classifier,#TestKnockoffNets#,142

Before Change

        // Create random attack
        attack = KnockoffNets(classifier=victim_ptc, batch_size_fit=BATCH_SIZE, batch_size_query=BATCH_SIZE,
                              nb_epochs=NB_EPOCHS, nb_stolen=NB_STOLEN, sampling_strategy="random")
        self.x_train = np.swapaxes(self.x_train, 1, 3)
        thieved_ptc = attack.extract(x=self.x_train, thieved_classifier=thieved_ptc)

        victim_preds = np.argmax(victim_ptc.predict(x=self.x_train), axis=1)
        thieved_preds = np.argmax(thieved_ptc.predict(x=self.x_train), axis=1)

After Change

        self.assertGreater(acc, 0.4)

        self.x_train_mnist = np.reshape(self.x_train_mnist, (self.x_train_mnist.shape[0], 28, 28, 1)).astype(np.float32)


class TestKnockoffNetsVectors(TestBase):

Instances