สร้าง Decision Tree

เมื่อแบ่งข้อมูลเที่ยวบินออกเป็นชุดฝึกและชุดทดสอบแล้ว ให้นำชุดฝึกไปใช้เทรนโมเดล Decision Tree

ข้อมูลพร้อมใช้งานในชื่อ flights_train และ flights_test

หมายเหตุ: โมเดลอาจใช้เวลาสักครู่ในการเทรน โปรดรอสักครู่!

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

Machine Learning with PySpark

ดูคอร์ส

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

Import คลาสสำหรับสร้าง Decision Tree classifier
สร้างออบเจกต์ classifier แล้ว fit กับข้อมูลชุดฝึก
ทำนายผลสำหรับข้อมูลชุดทดสอบ แล้วตรวจดูผลลัพธ์ที่ได้

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

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

# Import the Decision Tree Classifier class
from pyspark.ml.____ import ____

# Create a classifier object and fit to the training data
tree = ____()
tree_model = tree.____(____)

# Create predictions for the testing data and take a look at the predictions
prediction = tree_model.____(____)
prediction.select('label', 'prediction', 'probability').show(5, False)

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