我将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”