Bây giờ, bạn sẽ thử xây dựng một mô hình cây quyết định. Cây quyết định là một danh sách các quy tắc if-else được học bằng machine learning để quyết định — trong bài toán churn của viễn thông — liệu khách hàng có rời bỏ hay không. Dưới đây là một ví dụ đồ thị cây quyết định được xây trên bộ dữ liệu sống sót của Titanic nổi tiếng.

Các biến train_X, test_X, train_Y, test_Y từ bài tập trước đã được nạp sẵn cho bạn. Ngoài ra, mô-đun tree và hàm accuracy_score cũng đã được nạp từ thư viện sklearn. Giờ bạn sẽ xây dựng mô hình và kiểm tra hiệu năng của nó trên dữ liệu chưa thấy trước đó.

Vì sao dùng ML cho marketing? Chiến lược và bài toán ứng dụng

Create Your Free Account