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.
Bài tập này là một phần của khóa học
Phân tích Nhân sự (HR Analytics): Dự đoán nghỉ việc bằng Python
Hướng dẫn bài tập
- Import hàm
export_graphviz()từ phân hệ consklearn.tree. - Fit
modelvới dữ liệu huấn luyện. - Xuất trực quan hóa ra tệp
tree.dot.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# Import the graphical visualization export function
from sklearn.____ import export_graphviz
# Apply Decision Tree model to fit Features to the Target
model.____(features_train,target_train)
# Export the tree to a dot file
____(model,"tree.____")