套索回归预测指标

2024-05-03 23:52:15 发布

您现在位置:Python中文网/ 问答频道 /正文

我在同一个数据集上建立了一个岭和套索回归模型,但是套索模型的预测形状似乎与岭不同。如果有人能解释我做错了什么我会很感激。。。你知道吗

###Ridge Regression
model3 = Ridge(alpha=5)
model3.fit(x_train2, y_train)

y_pred3=model3.predict(x_test2)
y_pred3.shape
#result is: (1542, 1)

###Lasso Regression
from sklearn.linear_model import Lasso

model4 = Lasso(alpha=0.1)
model4.fit(x_train2, y_train)

y_pred4=model4.predict(x_test2)
y_pred4.shape
#result is: (1542,)

Tags: 模型alphatrainpredictfit套索shaperidge