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

Huấn luyện bộ phân loại AdaBoost

Bây giờ bạn đã khởi tạo bộ phân loại AdaBoost ada, đã đến lúc huấn luyện nó. Bạn cũng sẽ dự đoán xác suất thuộc lớp dương trên tập kiểm tra. Bạn có thể thực hiện như sau:

Sau khi huấn luyện xong bộ phân loại ada, gọi phương thức .predict_proba() với X_test làm tham số và trích xuất các xác suất này bằng cách lấy toàn bộ giá trị ở cột thứ hai như sau:

ada.predict_proba(X_test)[:,1]

Bộ dữ liệu Indian Liver đã được xử lý sẵn và 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 nhãn 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 mô hình ada đã khởi tạo từ bài tập trước.

Hướng dẫn

100 XP

  • Fit ada trên tập huấn luyện.

  • Tính xác suất thuộc lớp dương trên tập kiểm tra.