如何将机器学习（Tensorflow）预测导出到csv文件？

Question

我已将我的代码here添加到Google文档中。

我无法导出作为输出打印到csv文件的预测值。

for i in range(0, 101):
prediction_run = sess.run(prediction, feed_dict={x: X[i].reshape(1, 60)})
accuracy_run = sess.run(accuracy, feed_dict={x: X[i].reshape(1, 60), y_: Y[i].reshape(1, 2)})
print("Original Class : ", y1[i], " Predicted Values : ", prediction_run[0], " Accuracy : ", accuracy_run)

Answer 1

查看您的代码，您根本不使用pandas将其导出到csv。你能做的就是这样做：

import pandas as pd

prediction_value = sess.run(prediction, feed_dict={x: x_test}
prediction_df = pd.DataFrame(prediction_value)
prediction_df.to_csv('prediction.csv')

顺便说一句，下次最好将你的代码放在gist上而不是google驱动器中。