def lasso_train(X, y):
model_lasso = LassoCV(alphas = [1, 0.1, 0.001, 0.0005]).fit(X, y)
print ('lasso mean cv is ',rmse_cv(model_lasso,X,y).mean())
return model_lasso
#%%
微信
公众号
扫码关注公众号