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

Khởi tạo mô hình

Trong loạt bài tập sau, bạn sẽ chẩn đoán vấn đề bias và variance của một cây hồi quy. Cây hồi quy bạn sẽ định nghĩa trong bài này được dùng để dự đoán mức tiêu thụ mpg của ô tô từ bộ dữ liệu auto, sử dụng tất cả các đặc trưng sẵn có.

Chúng tôi đã xử lý dữ liệu và nạp sẵn ma trận đặc trưng X và mảng y vào không gian làm việc của bạn. Ngoài ra, lớp DecisionTreeRegressor đã được import từ sklearn.tree.

Hướng dẫn

100 XP
  • Import train_test_split từ sklearn.model_selection.
  • Chia dữ liệu thành 70% train và 30% test.
  • Khởi tạo DecisionTreeRegressor với độ sâu tối đa là 4 và min_samples_leaf đặt thành 0.26.