python - 在 python 中使用 fit 方法时数组的形状应该是什么？

Question

我是数据科学的新手。我正在使用 Spyder (Python 3.7)。我的数据集包含两列。一个是员工的经验年限（自变量），另一个是他们各自的薪水（因变量）。我将数据集分成训练集和测试集。LinearRegression之后，在使用库的类将简单的线性回归模型拟合到我的数据集中sklearn.linear_model，并使用regression.fit(x_train, y_train)时，我得到了错误，它说Expected 2D array, got 1D array instead。之后，它在 IPython 控制台中打印了 x_train 数组。它进一步说Reshape your data using array.reshape(-1, 1) if your data has a single feature or array.reshape(1, -1) if it contains a single sample 我不明白该怎么做。我还使用 将np.reshape()我的两个数组重塑为 (10, 2) 的形状，使它们成为二维数组，但仍然出现相同的错误。

标签： pythonarraysnumpylinear-regressiondata-science