c0e9a520c944e971b0ea53a9e81142e2e6bbab64,art/attacks/universal_perturbation_unittest.py,TestUniversalPerturbation,test_krclassifier,#TestUniversalPerturbation#,106

Before Change

        
        // Initialize a tf session
        session = tf.Session()
        k.set_session(session)

        // Get MNIST
        (x_train, y_train), (x_test, y_test) = self.mnist

        // Create simple CNN
        model = Sequential()
        model.add(Conv2D(4, kernel_size=(5, 5), activation="relu", input_shape=(28, 28, 1)))
        model.add(MaxPooling2D(pool_size=(2, 2)))
        model.add(Flatten())
        model.add(Dense(10, activation="softmax"))

        model.compile(loss=keras.losses.categorical_crossentropy, optimizer=keras.optimizers.Adam(lr=0.01),
                      metrics=["accuracy"])

        // Get classifier
        krc = KerasClassifier((0, 1), model, use_logits=False)
        krc.fit(x_train, y_train, batch_size=BATCH_SIZE, nb_epochs=2)

After Change

        :return:
        
        // Build KerasClassifier
        krc, sess = get_classifier_kr()

        // Get MNIST
        (x_train, y_train), (x_test, y_test) = self.mnist

Instances