Tensorflowのテストデータをバッチではなく１つ１つ確認したい

Question

Tensorflowで、一般的なバッチでテストデータの精度を算出しているので以下のようにして実行できています。

test_acc = []
with tf.Session(graph=graph) as sess:
    saver.restore(sess, tf.train.latest_checkpoint('checkpoint'))
    test_state = sess.run(cell.zero_state(batch_size, tf.float32))
    for ii, (x,y) in enumerate(get_batches(test_x, test_y, batch_size), 1):
        feed = {inputs_: x,
               labels_: y[:,None],
               keep_prob: 1,
               initial_state: test_state}
        batch_acc, test_state = sess.run([accuracy, final_state], feed_dict=feed)
        test_acc.append(batch_acc)
    print("Test Accuracy: {:.3f}".format(np.mean(test_acc)))

test_x.shape
(2500, 200)

test_y.shape
(2500,)

今回は2クラス分類を行っています。

しかし、test_xの中の２、３個抽出して、抽出したデータがどのクラスに属するかを知りたい場合どのようにしたら求めることができるでしょうか。

よろしくお願い致します。

Answer 1

計算グラフ accuracy で精度を計算しているということは、その手前に推論した確率を出力しているノードがあると思うので、そのノードを sess.run() すればいいです。