python - 如何将机器学习 (Tensorflow) 预测导出到 csv 文件？

Question

我已经包含了我的代码 here在谷歌文档上。

我无法将作为输出打印的预测值导出到 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 上而不是谷歌驱动器中。