1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Machine Learning với mô hình dựa trên cây trong Python
Connected

Bài tập

Đánh giá bộ hồi quy RF

Giờ bạn sẽ đánh giá RMSE trên tập kiểm tra của bộ hồi quy random forests rf mà bạn đã huấn luyện ở bài trước.

Bộ dữ liệu đã được xử lý sẵn và chia thành 80% train và 20% test. Ma trận đặc trưng X_test và mảng y_test có sẵn trong không gian làm việc của bạn. Ngoài ra, chúng tôi cũng đã nạp mô hình rf mà bạn đã huấn luyện ở bài trước.

Hướng dẫn

100 XP
  • Import mean_squared_error từ sklearn.metrics với bí danh MSE.
  • Dự đoán nhãn cho tập kiểm tra và gán kết quả vào y_pred.
  • Tính RMSE trên tập kiểm tra và gán vào rmse_test.