import tensorflow as tf

                         reduction_indices=1)
# Prediction: Get min distance index (Nearest neighbor)
pred = tf.arg_min(distance, 0)

accuracy = 0.

# Initialize the variables (i.e. assign their default value)
init = tf.global_variables_initializer()

# Start training
with tf.compat.v1.Session() as sess:

    # Run the initializer
    sess.run(init)

    # Add the fault injection code here to instrument the graph
    # We start injecting the fault right away here unlike earlier
    fi = ti.TensorFI(sess, name="NearestNeighbor", logLevel=50)

    # loop over test data
