1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Machine Learning với mô hình dựa trên cây trong Python
Connected

Bài tập

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.

Hướng dẫn

100 XP
  • 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.