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()

您正在绘制来自“regr.fit”的“diabetes_y_pred”，而 regr 在上面用“LinearRegression()”定义。

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