การ Export ต้นไม้การตัดสินใจ

ในงาน classification ด้วย Decision Tree นั้น overfitting มักเกิดจากการที่ต้นไม้ถูกสร้างให้ลึกเกินไป จากการเปรียบเทียบคะแนน accuracy บนชุดข้อมูล train และ test จะเห็นได้ว่าผลลัพธ์ที่ได้มี overfitting เกิดขึ้น ซึ่งสังเกตได้จากการแสดงผลต้นไม้เช่นกัน

ในแบบฝึกหัดนี้ จะได้ export decision tree ออกเป็นเอกสารข้อความ เพื่อนำไปใช้ในการแสดงภาพต่อไป

แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร

HR Analytics: การพยากรณ์การลาออกของพนักงานด้วย Python

ดูคอร์ส

คำแนะนำการฝึกหัด

Import ฟังก์ชัน export_graphviz() จาก submodule sklearn.tree
Fit model กับข้อมูล training
Export ภาพต้นไม้ออกเป็นไฟล์ tree.dot

แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ

ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์

# 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.____")

แก้ไขและรันโค้ด