Huấn luyện cây hồi quy đầu tiên của bạn
Trong bài tập này, bạn sẽ huấn luyện một cây hồi quy để dự đoán mức tiêu thụ mpg (miles per gallon) của ô tô trong bộ dữ liệu auto-mpg bằng cách sử dụng cả sáu đặc trưng sẵn có.
Bộ dữ liệu đã được xử lý sẵn và được chia 80% train và 20% test. Ma trận đặc trưng X_train và mảng y_train đã có trong không gian làm việc của bạn.
Bài tập này là một phần của khóa học
Machine Learning với mô hình dựa trên cây trong Python
Hướng dẫn bài tập
- Import
DecisionTreeRegressortừsklearn.tree. - Khởi tạo một
DecisionTreeRegressordtvới độ sâu tối đa là 8 vàmin_samples_leafđặt là 0.13. - Fit
dttrên tập huấn luyệ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 DecisionTreeRegressor from sklearn.tree
from ____.____ import ____
# Instantiate dt
____ = ____(max_depth=____,
____=____,
random_state=3)
# Fit dt to the training set
____.____(____, ____)