CODEWITHSUNDEEP
pythoncurve-fitting
satya
satya

Reputation: 1

Polynomial fitting in Python

i am trying to do a polynomial fit to my data and the output is a linear fit. i am trying to understand where my mistake is.

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)

Upvotes: 0

Views: 156

Answers (1)

angrymantis
angrymantis

Reputation: 382

It's doing it because you told it to.

regr = LinearRegression()

You are plotting "diabetes_y_pred," which comes from "regr.fit" and regr is defined above with "LinearRegression()."

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

Upvotes: 1

Related Questions