Regresión polinómica en python

Question

¿Cómo implemento la regresión polinómica en Python? ¿Utilizo simplemente el LinearRegression o hay alguna biblioteca especial que pueda utilizar?

Entiendo que es una pregunta muy amplia, pero no he podido encontrar ninguna implementación clara en una simple búsqueda en Google.

Si alguien ha implementado la regresión polinómica en python antes, la ayuda sería muy apreciada

Gracias.

Answer 1

Comprobar la regresión polinómica implementada con sklearn aquí .

Si conoce la regresión lineal, la regresión polinómica es casi lo mismo, excepto que usted elige el grado del polinomio y lo convierte en una forma adecuada para que el regresor lineal lo utilice después.

from sklearn.preprocessing import PolynomialFeatures
from sklearn.linear_model import LinearRegression

X_train = [[1,2],[2,4],[6,7]]
y_train = [1.2, 4.5, 6.7]
X_test = [[1,3],[2,5]]    

# create a Linear Regressor   
lin_regressor = LinearRegression()

# pass the order of your polynomial here  
poly = PolynomialFeatures(2)

# convert to be used further to linear regression
X_transform = poly.fit_transform(X_train)

# fit this to Linear Regressor
lin_regressor.fit(X_transform,y_train) 

# get the predictions
X_test_transform = poly.fit_transform(X_test)
y_preds = lin_regressor.predict(X_test_transform)