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

Dùng entropy làm tiêu chí

Trong bài tập này, bạn sẽ huấn luyện một cây phân loại trên tập dữ liệu Wisconsin Breast Cancer, dùng entropy làm tiêu chí thông tin. Bạn sẽ sử dụng toàn bộ 30 đặc trưng trong tập dữ liệu, với tỷ lệ chia 80% train và 20% test.

X_train cũng như mảng nhãn y_train đã có sẵn trong không gian làm việc của bạn.

Hướng dẫn

100 XP

  • Import DecisionTreeClassifier từ sklearn.tree.

  • Khởi tạo DecisionTreeClassifier dt_entropy với độ sâu tối đa là 8.

  • Đặt tiêu chí thông tin là 'entropy'.

  • Fit dt_entropy trên tập huấn luyện.