1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Phân tích Nhân sự (HR Analytics): Dự đoán nghỉ việc bằng Python
Connected

Bài tập

Xuất cây

Trong các bài toán phân loại bằng Decision Tree, overfitting thường xảy ra khi cây phát triển quá sâu. Như so sánh điểm chính xác giữa tập huấn luyện và tập kiểm tra cho thấy, kết quả của bạn đang bị overfitting. Điều này cũng có thể nhận ra qua trực quan hóa của cây.

Trong bài tập này, bạn sẽ xuất cây quyết định ra một tài liệu văn bản, sau đó có thể dùng để trực quan hóa.

Hướng dẫn

100 XP
  • Import hàm export_graphviz() từ phân hệ con sklearn.tree.
  • Fit model với dữ liệu huấn luyện.
  • Xuất trực quan hóa ra tệp tree.dot.