我尝试创建一个DataFrame以便于可视化真实和预测比较。但是使用代码:
pred = pd.DataFrame({'Реальная': y_test,
'Предсказанная': model.predict(X_test)})
我明白了:
Exception: Data must be 1-dimensional
因此,您可以转到My Colab,而不必注意俄语注释(请确保您是нерусские))
答案 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