1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Gradient Boosting Cực Mạnh với XGBoost
Connected

Bài tập

Random search với XGBoost

Thường thì GridSearchCV rất tốn thời gian, nên trong thực tế bạn có thể muốn dùng RandomizedSearchCV thay thế, như bạn sẽ làm trong bài này. Tin tốt là bạn chỉ cần chỉnh sửa một chút từ mã GridSearchCV để dùng RandomizedSearchCV. Khác biệt quan trọng là bạn phải truyền tham số param_distributions thay vì param_grid.

Hướng dẫn

100 XP
  • Tạo lưới tham số tên gbm_param_grid gồm một danh sách với một giá trị duy nhất cho 'n_estimators' (25), và một danh sách các giá trị 'max_depth' từ 2 đến 11 cho 'max_depth' - dùng range(2, 12) cho phần này.
  • Tạo một đối tượng RandomizedSearchCV tên randomized_mse, truyền vào: lưới tham số cho param_distributions, XGBRegressor cho estimator, "neg_mean_squared_error" cho scoring, 5 cho n_iter, và 4 cho cv. Đồng thời chỉ định verbose=1 để bạn dễ hiểu đầu ra hơn.
  • Fit đối tượng RandomizedSearchCV với X và y.