套索回归预测忽略指数

时间:2018-01-06 17:51:15

标签: python arrays scikit-learn prediction lasso

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

###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,)

1 个答案:

答案 0 :(得分:0)

我仍然不知道为什么Lasso模型的预测形状不同但我通过将数组重塑为:

来解决我的问题
mongo