来自Keras预测的数据的pd.DataFrame问题

时间:2019-07-13 18:23:43

标签: python pandas numpy keras

我尝试创建一个DataFrame以便于可视化真实和预测比较。但是使用代码:

pred = pd.DataFrame({'Реальная': y_test, 
                 'Предсказанная': model.predict(X_test)})

我明白了:

Exception: Data must be 1-dimensional

因此,您可以转到My Colab,而不必注意俄语注释(请确保您是нерусские))

1 个答案:

答案 0 :(得分:1)

尝试展平并将其转换为列表

pred = pd.DataFrame({'Реальная': y_test, 
                     'Предсказанная': list(model.predict(X_test).flatten())})
print(pred)
      Реальная  Предсказанная
0         7.2       5.699125
1        18.8      19.322182
2        19.0      23.012312
3        27.0      30.977814
4        22.2      25.090652