Tạo một bộ hồi quy LassoCV
Bạn sẽ dự đoán chu vi bắp tay trên một mẫu con của bộ dữ liệu ANSUR dành cho nam, sử dụng bộ hồi quy LassoCV() tự động tinh chỉnh độ mạnh regularization (giá trị alpha) bằng Cross-Validation.
Dữ liệu huấn luyện và kiểm tra đã được chuẩn hóa được nạp sẵn dưới dạng X_train, X_test, y_train, và y_test.
Bài tập này là một phần của khóa học
Giảm Chiều Dữ Liệu với Python
Hướng dẫn bài tập
- Tạo và fit mô hình LassoCV trên tập huấn luyện.
- Tính \(R^2\) trên tập kiểm tra.
- Tạo một mask cho các hệ số khác 0.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
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')