Python中的多项式拟合

Question

我试图对我的数据进行多项式拟合，结果是线性拟合。我正在努力理解我的错误在哪里

regr2 = PolynomialFeatures(degree=2)
regr2.fit_transform(diabetes_X_train, diabetes_y_train)
regr2 = PolynomialFeatures(interaction_only=True)
regr2.fit_transform(diabetes_X_train, diabetes_y_train)

regr = LinearRegression()
regr.fit(diabetes_X_train, diabetes_y_train)

diabetes_y_pred = regr.predict(diabetes_X_test)

plt.scatter(diabetes_X_test, diabetes_y_test,  color='blue')
plt.plot(diabetes_X_test, diabetes_y_pred, color='red', linewidth=1)

Answer 1

它这么做是因为你让它这么做

regr = LinearRegression()

您正在绘制“糖尿病y_pred”，它来自“regr.fit”，而regr在上面是用“LinearRegression（）定义的

https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LinearRegression.html