テンソルフロー予測結果を.csvで出力する方法は？

Question

私はCNNを使用してモデルをトレーニングしています。

ここに私の予測部分があります。

predictions = { 
    "classes": tf.argmax(input=logit2, axis=1), 
    "probabilities": tf.nn.softmax(logit2, name="softmax_tensor") 
} 

ここに評価を行うコードがあります。

eval_input_fn = tf.estimator.inputs.numpy_input_fn(
    x={"x": images_test}, 
    y=test_labels, 
    num_epochs=1, 
    shuffle=False) 
eval_results = model.evaluate(input_fn=eval_input_fn) 

私は（csvファイルの最初の列に）テスト画像名のリストを持っている、と私は予測して出力を二列に対応する結果を（作りたい今、私のモデルを訓練してきました確率は0と1の間で）、これを達成する方法、およびコードを追加する場所

ありがとうございます。

Answer 1

Estimatorクラスには、予測を反復可能オブジェクト（例：scroll to the very bottom of this page）として返すpredict functionがあります。

ので、あなたができる：CSVファイルの2列目への書き込みについては

predictions = model.predict(input_fn=predict_input_fn) 
for p in predictions: 
    # write p['classes'] to the csv 

を、csvのpythonモジュールを見てみましょう。