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

Bài tập

Phân rã các luật của cây quyết định

Trong bài tập này, bạn sẽ trích xuất các quy tắc if-else từ cây quyết định và vẽ chúng để xác định những yếu tố chính dẫn đến churn.

Thực thể cây quyết định đã được huấn luyện được nạp dưới tên mytree và các đặc trưng đã được scale được nạp dưới dạng pandas DataFrame tên train_X. Mô-đun tree từ thư viện sklearn và thư viện graphviz cũng đã được nạp sẵn cho bạn.

Lưu ý: chúng tôi dùng hàm độc quyền display_image() thay cho display(graph) để bạn xem đầu ra dễ hơn.

Hướng dẫn

100 XP
  • Xuất đối tượng graphviz từ cây quyết định đã huấn luyện.
  • Gán tên các đặc trưng.
  • Đặt precision = 1 và thêm tên các lớp.
  • Gọi hàm Source() từ graphviz và truyền vào đối tượng graphviz đã xuất.