x_train = np.array([6, 8, 10, 14, 18])
y_ = np.array([7, 9, 13, 17.5, 18])
model.fit(x_train, y_)
错误
ValueError: Expected 2D array, got 1D array instead
期望 2 维数组,输入的是一维数组
Reshape your data either using array.reshape(-1, 1) if your data has a single feature or array.reshape(1, -1) if it contains a single sample.
这是由于在sklearn中,所有的数据都应该是二维矩阵,哪怕它只是单独一行或一列(比如前面做预测时,仅仅只用了一个样本数据),所以需要使用.reshape(1,-1)
进行转换
解决
x_train = np.array([6, 8, 10, 14, 18]).reshape(-1, 1)
y_ = np.array([7, 9, 13, 17.5, 18]).reshape(-1, 1)
model.fit(x_train, y_)