From 5d896bf7141c367a9e807025cff2c237a1a66df3 Mon Sep 17 00:00:00 2001 From: jason_cv Date: Tue, 14 Apr 2020 17:43:29 -0500 Subject: [PATCH] Modelo lineal regularizado Ridge --- RegresionRidge.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 RegresionRidge.py diff --git a/RegresionRidge.py b/RegresionRidge.py new file mode 100644 index 0000000..46ccda7 --- /dev/null +++ b/RegresionRidge.py @@ -0,0 +1,21 @@ +from sklearn.linear_model import Ridge +from sklearn.preprocessing import PolynomialFeatures +import numpy as np +import matplotlib.pyplot as plt +############################### +#Datos originales +############################### +m = 100 +X = 6 * np.random.rand(m, 1) - 3 +y = 0.5 * X**2 + X + 2 + np.random.randn(m, 1) + +plt.plot(X,y,".") +############################### +poly_features = PolynomialFeatures(degree=2, include_bias=False) +X_pol = poly_features.fit_transform(X) +ridge_reg = Ridge(alpha=1, solver="cholesky") +ridge_reg.fit(X_pol, y) +yout=ridge_reg.predict(X_pol) +print(ridge_reg.predict(1.5,2)) +plt.plot(X,yout,"*") +plt.show() \ No newline at end of file