开始使用免费开始使用

创建 LassoCV 回归器

您将使用 LassoCV() 回归器在男性 ANSUR 数据集的一个子样本上预测肱二头肌围度。该回归器会通过交叉验证自动调节正则化强度(alpha 值)。

已为您预先加载了标准化后的训练集与测试集,变量分别为 X_trainX_testy_trainy_test

本练习是课程的一部分

Python 中的降维

查看课程

练习说明

  • 在训练集上创建并拟合 LassoCV 模型。
  • 在测试集上计算 $R^2$。
  • 为不等于 0 的系数创建一个掩码。

交互式实操练习

通过完成这段示例代码来试试这个练习。

from sklearn.linear_model import LassoCV

# Create and fit the LassoCV model on the training set
lcv = ____
lcv.____
print(f'Optimal alpha = {lcv.alpha_:.3f}')

# Calculate R squared on the test set
r_squared = lcv.____
print(f'The model explains {r_squared:.1%} of the test set variance')

# Create a mask for coefficients not equal to zero
lcv_mask = ____
print(f'{sum(lcv_mask)} features out of {len(lcv_mask)} selected')
编辑并运行代码