Hồi quy tuyến tính so với cây hồi quy
Trong bài tập này, bạn sẽ so sánh RMSE trên tập kiểm tra của dt với mô hình hồi quy tuyến tính. Chúng tôi đã khởi tạo sẵn mô hình hồi quy tuyến tính lr và huấn luyện nó trên cùng bộ dữ liệu với dt.
Ma trận đặc trưng X_test, mảng nhãn y_test, mô hình hồi quy tuyến tính đã huấn luyện lr, hàm mean_squared_error được import với bí danh MSE và rmse_dt từ bài trước đều có sẵn trong không gian làm việc của bạn.
Bài tập này là một phần của khóa học
Machine Learning với mô hình dựa trên cây trong Python
Hướng dẫn bài tập
Dự đoán nhãn cho tập kiểm tra bằng mô hình hồi quy tuyến tính (
lr) và gán kết quả choy_pred_lr.Tính MSE trên tập kiểm tra và gán kết quả cho
mse_lr.Tính RMSE trên tập kiểm tra và gán kết quả cho
rmse_lr.
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.
# Predict test set labels
____ = ____.____(____)
# Compute mse_lr
____ = ____(____, ____)
# Compute rmse_lr
____ = ____
# Print rmse_lr
print('Linear Regression test set RMSE: {:.2f}'.format(rmse_lr))
# Print rmse_dt
print('Regression Tree test set RMSE: {:.2f}'.format(rmse_dt))