我正在将 sklearn 回归模型的结果存储到变量预测中。
prediction = regressor.predict(data[['X']])
print(prediction)
预测输出的值如下所示
[ 266.77832991 201.06347505 446.00066136 499.76736079 295.15519906
214.50514991 422.1043505 531.13126879 287.68760191 201.06347505
402.68859792 478.85808879 286.19408248 192.10235848]
然后我尝试使用 to_csv 函数将结果保存到本地 CSV 文件:
prediction.to_csv('C:/localpath/test.csv')
但是我得到的错误是:
AttributeError: 'numpy.ndarray' object has no attribute 'to_csv'
我正在使用 Pandas/Numpy/SKlearn。对基本修复有任何想法吗？
最佳答案
你可以使用 Pandas 。 正如所说，numpy 数组没有 to_csv 函数。
import numpy as np
import pandas as pd
prediction = pd.DataFrame(predictions, columns=['predictions']).to_csv('prediction.csv')
如果您想要行内或列状的值，请添加“.T”。
关于python - 将预测结果保存到 CSV，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/34864695/