tensorflow : Export predictions to an array or a file

Question

由于所有计算都是在 session 下进行的，有没有办法将 Tensorflow 的预测导出到 Numpy/Pandas 数组或文件，即 CSV 或 TXT ？

谢谢 !
保罗

Answer 1

你想使用类似...

im = Image.open('/home/kendall/Desktop/HA900_frames/frame0635.tif')
batch_x = np.array(im)
batch_x = batch_x.reshape((1, n_steps, n_input))
batch_x = batch_x.astype(float)
prediction = sess.run(pred, feed_dict={x: batch_x})
prediction = np.asarray(prediction)
prediction = np.reshape(prediction, (200,200))
np.savetxt("prediction.csv", prediction, delimiter=",")