Xây dựng Decision Tree
Giờ bạn đã tách dữ liệu chuyến bay thành tập huấn luyện và tập kiểm tra, bạn có thể dùng tập huấn luyện để huấn luyện một mô hình Decision Tree.
Dữ liệu có sẵn dưới dạng flights_train và flights_test.
LƯU Ý: Mô hình sẽ mất vài giây để huấn luyện… vui lòng chờ một chút!
Bài tập này là một phần của khóa học
Machine Learning với PySpark
Hướng dẫn bài tập
- Import lớp để tạo bộ phân loại Decision Tree.
- Tạo một đối tượng bộ phân loại và huấn luyện nó trên dữ liệu huấn luyện.
- Tạo dự đoán cho dữ liệu kiểm tra và xem thử các dự đoán.
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 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)