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á rừng tối ưu

Trong bài tập cuối cùng của khóa học, bạn sẽ đánh giá RMSE trên tập kiểm tra của mô hình tối ưu trong grid_rf.

Bộ dữ liệu đã được tải và xử lý sẵn, đồng thời được chia 80% huấn luyện và 20% kiểm tra. Trong môi trường của bạn có sẵn X_test, y_test và hàm mean_squared_error từ sklearn.metrics với bí danh MSE. Ngoài ra, chúng tôi cũng đã tải đối tượng GridSearchCV đã huấn luyện là grid_rf mà bạn đã khởi tạo ở bài trước. Lưu ý rằng grid_rf đã được huấn luyện như sau:

grid_rf.fit(X_train, y_train)

Hướng dẫn

100 XP

  • Import mean_squared_error dưới bí danh MSE từ sklearn.metrics.

  • Trích xuất bộ ước lượng tốt nhất từ grid_rf và gán cho best_model.

  • Dự đoán nhãn tập kiểm tra bằng best_model và gán kết quả cho y_pred.

  • Tính RMSE trên tập kiểm tra của best_model.