Как я могу сохранить предсказанные значения 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.Планшеты данных Pandas - это в основном таблицы, имеющие более одного столбца.Поскольку переменная degrees является единственным значением, вам следует использовать здесь Pandas Series.Это должно работать:

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