134d00c07c5f2bfb6e79811e9397e5a111635f91,art/attacks/spatial_transformation_unittest.py,TestSpatialTransformation,test_tfclassifier,#TestSpatialTransformation#,230

Before Change

        
        // Build a TFClassifier
        // Define input and output placeholders
        input_ph = tf.placeholder(tf.float32, shape=[None, 28, 28, 1])
        output_ph = tf.placeholder(tf.int32, shape=[None, 10])

        // Define the tensorflow graph
        conv = tf.layers.conv2d(input_ph, 1, 7, activation=tf.nn.relu, kernel_initializer=tf_initializer_w_conv2d,
                                bias_initializer=tf_initializer_b_conv2d)
        conv = tf.layers.max_pooling2d(conv, 4, 4)
        flattened = tf.contrib.layers.flatten(conv)

        // Logits layer
        logits = tf.layers.dense(flattened, 10, kernel_initializer=tf_initializer_w_dense,
                                 bias_initializer=tf_initializer_b_dense)

        // Train operator
        loss = tf.reduce_mean(tf.losses.softmax_cross_entropy(logits=logits, onehot_labels=output_ph))

        // Tensorflow session and initialization
        sess = tf.Session()
        sess.run(tf.global_variables_initializer())

After Change

        :return:
        
        // Build TF model
        loss, logits, input_ph, output_ph = get_model_tf()

        // Tensorflow session and initialization
        sess = tf.Session()

Instances