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.
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
DecisionTreeClassifiertừsklearn.tree.Khởi tạo
DecisionTreeClassifierdt_entropyvới độ sâu tối đa là 8.Đặt tiêu chí thông tin là
'entropy'.Fit
dt_entropytrê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 DecisionTreeClassifier from sklearn.tree
from ____.____ import ____
# Instantiate dt_entropy, set 'entropy' as the information criterion
dt_entropy = ____(____=____, ____='____', random_state=1)
# Fit dt_entropy to the training set
____.____(____, ____)