1. Learn
    2. /
  2. Courses
    3. /
  3. Ensemble Methods in Python
Connected

Exercise

Dự đoán doanh thu phim với CatBoost

Hãy kết thúc chương về boosting bằng cách quay lại bộ dữ liệu movies! Trong bài tập này, bạn sẽ xây dựng một CatBoostRegressor để dự đoán log-revenue. Nhớ rằng mô hình tốt nhất cho đến giờ là AdaBoost với RMSE bằng 5.15.

Liệu CatBoost có vượt qua AdaBoost? Chúng ta sẽ thử dùng một bộ tham số tương tự để so sánh công bằng.

Nhắc lại, đây là các đặc trưng chúng ta đã dùng: 'budget', 'popularity', 'runtime', 'vote_average', và 'vote_count'. catboost đã được nhập sẵn với tên cb.

LƯU Ý: đừng dùng classifier, nếu không phiên làm việc của bạn có thể bị hết hạn!

Instructions

100 XP
  • Xây dựng và fit một CatBoostRegressor với 100 estimators, learning rate 0.1, và max depth 3.
  • Tính dự đoán cho tập kiểm tra và in ra RMSE.