如何保存Keras预测值？

Question

我想将keras模型的预测转向角值保存在csv或txt文件中。

我想用它来绘制线图。我试图在不保存值的情况下进行绘制，但是看起来很复杂。

我用熊猫来保存值，但收到错误消息。

我写道：

degrees = float(model.predict(image, batch_size=1))
degrees = pd.DataFrame(degrees, columns=['pred1']).to_csv('data1.csv')

我收到了此消息。

ValueError: DataFrame constructor not properly called!

如何正确保存预测值？

Answer 1

根据我对您的问题的理解，您的模型返回的是一个要转换为float的值。熊猫数据框基本上是具有多列的表。由于degrees变量是单个值，因此您应在此处使用Pandas Series。这应该起作用：

degrees = float(model.predict(image, batch_size=1))
degrees = pd.Series(degrees).to_csv('data1.csv', header='pred1')