Skip to main content
 首页 » 编程设计

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

2025年02月15日31jirigala

我已经包含了我的代码 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) 

请您参考如下方法:

查看您的代码,您根本没有使用 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 上而不是谷歌驱动器中。