将预测结果保存到CSV

我将sklearn回归模型的结果存储到varibla预测中

prediction=regressor.predict(数据[['X']]
打印(预测)

预测输出的值如下所示

[266.77832991201.06347505446.00066136499.76736079295.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'对象没有“to_csv”属性

我正在使用Pandas/Numpy/SKlearn。对基本修复有什么想法吗

你可以用熊猫。
如前所述,numpy阵列没有to_csv函数

将numpy导入为np
作为pd进口熊猫
prediction=pd.DataFrame(预测,列=['predictions'])。到_csv('prediction.csv'))

如果您想让您的值以行或列的形式显示,请添加“.T”

发表评论