使用sklearn报错ValueError: Expected 2D array, got 1D array instead

1 问题描述

表述当前模型的输入应为二维数组，而得到的是一维数组

2 解决方法

2.1 使用array调整数据的形状，如果如果数据有单个功能或数组，则重新调整形状（-1，1）。如果数据包含单个示例，则重新调整形状（1，-1）。

new_x = np.array(new_x).reshape(1, -1)
pre_y = model.predict(new_x)

2.2 调整scikit-learn的版本

降低到稳定版本

pip install scikit-learn=0.15.0