2016-06-17 2 views
4

Как все вычисления сделаны в рамках сеанса, есть ли способ экспортировать предсказания Tensorflow в массив Numpy/Pandas или файл, то есть CSV или TXT?Tensorflow: экспортировать прогнозы в массив или файл

Спасибо! Paul

+2

Вы получаете результаты обратно в питона земли, так что вы должны быть в состоянии использовать любой пример питона. Я, вероятно, просто создаю объект «results» (Numpy/Pandas/list) и добавлю результаты от 'sess.run' к этому, и напишу его, как только вы закончите. –

+0

Контур с Джейкоб. Джейкоб, вы бы скопировали свою записку в ответ, чтобы я мог ее продвинуть и дать вам кредит? :) :) – dga

+0

Не знаю, почему я об этом не думал ... Спасибо! Paul –

ответ

2

Вы хотите использовать что-то вроде ...

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=",") 
-2

Вы можете использовать TFRecordWriter писать вывод в какой-то файл. https://www.tensorflow.org/api_docs/python/tf/python_io/TFRecordWriter

Приведенные ниже статьи могут дать вам представление о том, как это сделать: http://warmspringwinds.github.io/tensorflow/tf-slim/2016/12/21/tfrecords-guide/ http://machinelearninguru.com/deep_learning/tensorflow/basics/tfrecord/tfrecord.html

+0

Может оказаться полезным рабочий пример. – Lukas

Смежные вопросы