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

Connected

Bài tập

Huấn luyện GB regressor

Bây giờ bạn sẽ huấn luyện bộ hồi quy gradient boosting gb mà bạn đã khởi tạo ở bài trước và dự đoán nhãn cho tập kiểm tra.

Bộ dữ liệu được chia thành 80% huấn luyện và 20% kiểm tra. Ma trận đặc trưng X_train và X_test, cũng như các mảng y_train và 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 sẵn thể hiện mô hình gb mà bạn đã định nghĩa ở bài trước.

Hướng dẫn

100 XP
  • Fit gb lên tập huấn luyện.
  • Dự đoán nhãn của tập kiểm tra và gán kết quả cho y_pred.