Как напечатать вероятности прогноза в примере переподготовки Tensor Flow?

Question

Я использую пример Tensor Flow retraining в своем собственном наборе данных.Итоговая оценка теста выводит итоговую точность теста и имена неправильно классифицированных изображений:

test_accuracy, predictions = eval_session.run(
  [evaluation_step, prediction],
  feed_dict={
      bottleneck_input: test_bottlenecks,
      ground_truth_input: test_ground_truth
  })
  tf.logging.info('Final test accuracy = %.1f%% (N=%d)' %
              (test_accuracy * 100, len(test_bottlenecks)))

  if FLAGS.print_misclassified_test_images:
    tf.logging.info('=== MISCLASSIFIED TEST IMAGES ===')
    for i, test_filename in enumerate(test_filenames):
      if predictions[i] != test_ground_truth[i]:
        tf.logging.info('%70s  %s' % (test_filename, list(image_lists.keys())[predictions[i]]))

Как я могу также распечатать вероятности, связанные с прогнозами для всех классов?

Например:

image1 - A: 0,5;В: 0,3;С: 0,1;D: 0,1

изображение 2 - A: 0,3;В: 0,2;С: 0: 4;D: 0,1

Answer 1

Я думаю, что нашел ответ сам.

Вероятности можно получить так:

probs = tf.nn.softmax(final_tensor)
probabilities = sess.run(probs, feed_dict={bottleneck_input: test_bottlenecks,
    ground_truth_input: test_ground_truth})

Затем к ним можно получить доступ следующим образом:

for i, test_filename in enumerate(test_filenames):        
    tf.logging.info('%70s  %f %f %f %f' %
                      (test_filename,
                       probabilities[i][0], probabilities[i][1], probabilities[i][2], probabilities[i][3]))