python evaluation_function tensorflow

Posted 2021-05-09

# add to training file, after variables have been defined
tf.add_to_collection('inputs', x)
tf.add_to_collection('inputs', y)
tf.add_to_collection('inputs', one_hot_y)
tf.add_to_collection('outputs', logits)
tf.add_to_collection('outputs', correct_prediction)
tf.add_to_collection('outputs', accuracy_operation)

# in testing file
with tf.Graph().as_default() as g:
    with tf.Session() as sess:
        loader = tf.train.import_meta_graph('lenet.meta')
        x,y, one_hot= tf.get_collection('inputs')
        logits, correct_prediction, accuracy_operation = tf.get_collection('outputs')
        loader.restore(sess, tf.train.latest_checkpoint('./'))
        
        test_accuracy = evaluate(X_test, y_test, sess)
        print("Test Accuracy = {:.3f}".format(test_accuracy))