1. Học hỏi
  2. /
  3. Khoa Học
  4. /
  5. Machine Learning cho Marketing với Python

Connected

Bài tập

Huấn luyện mô hình cây quyết định

Bây giờ bạn sẽ huấn luyện một cây quyết định trên tập huấn luyện của bộ dữ liệu viễn thông, sau đó dự đoán nhãn trên dữ liệu kiểm tra chưa từng thấy và tính độ chính xác của dự đoán mô hình. Bạn sẽ thấy sự khác biệt về hiệu năng so với logistic regression.

Hàm accuracy_score đã được import, và các tập dữ liệu huấn luyện/kiểm tra mà bạn đã xây dựng trước đó cũng đã được nạp: train_X và test_X cho đặc trưng, train_Y và test_Y cho biến mục tiêu.

Hướng dẫn

100 XP
  • Khởi tạo một bộ phân loại cây quyết định.
  • Fit cây quyết định trên dữ liệu huấn luyện.
  • Dự đoán nhãn churn trên dữ liệu kiểm tra.
  • Tính và in ra accuracy score trên dữ liệu kiểm tra.