pycharm如何将模型的预测结果保存到CSV文件,并且加载使用?

1.预测结果数据如下图,仅一维

pycharm如何将模型的预测结果保存到CSV文件,并且加载使用?_第1张图片

 2.数据保存,因为预测好的数据是数组格式。先要转成表格格式,再保存到CSV文件中。

# 1.数据格式转换:数组(ndarray) 转成 表格(DataFrame)
# 前columns设置标签名(预测结果是两列数,就加两个标签),后data为待保存的数据
test=pd.DataFrame(columns=['1'], data=y_test_pred)

# 2.数据保存,index表示是否显示行名,sep数据分开符
test.to_csv('save1.csv', index=False, sep=',')

3.读取加载保存的实验结果数据,因为加载的数据是表格格式,要将其转成数组格式才能使用。

# 数据加载
df1 = pd.read_csv('save1.csv')
# 数据格式转换,表格(DataFrame) 转成 数组(ndarray)
data = np.array(df1)

4.效果如下:

pycharm如何将模型的预测结果保存到CSV文件,并且加载使用?_第2张图片

 

你可能感兴趣的:(深度学习,python,机器学习,开发语言)