Bắt đầu ngayBắt đầu miễn phí

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

Xem khóa học

Hướng dẫn bài tập

  • Import DecisionTreeRegressor từ sklearn.tree.
  • Khởi tạo một DecisionTreeRegressor dt với độ sâu tối đa là 8 và min_samples_leaf đặt là 0.13.
  • Fit dt trê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
____.____(____, ____)
Chỉnh sửa và Chạy Mã