ฝึก Regression Tree ต้นแรกของคุณ
ในแบบฝึกหัดนี้ จะได้ฝึก regression tree เพื่อพยากรณ์ค่า mpg (ไมล์ต่อแกลลอน) ของรถยนต์ใน ชุดข้อมูล auto-mpg โดยใช้ฟีเจอร์ทั้ง 6 ตัวที่มีอยู่
ชุดข้อมูลได้รับการเตรียมไว้ให้แล้ว และแบ่งเป็นชุด train 80% กับชุด test 20% โดย features matrix X_train และอาร์เรย์ y_train พร้อมใช้งานใน workspace ของคุณ
แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร
Machine Learning with Tree-Based Models in Python
คำแนะนำการฝึกหัด
- นำเข้า
DecisionTreeRegressorจากsklearn.tree - สร้าง instance ของ
DecisionTreeRegressorชื่อdtโดยกำหนด maximum depth เป็น 8 และmin_samples_leafเป็น 0.13 - Fit
dtกับชุดข้อมูล training
แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ
ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์
# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____
# Instantiate dt
____ = ____(max_depth=____,
____=____,
random_state=3)
# Fit dt to the training set
____.____(____, ____)