Huấn luyện cây trên dữ liệu nhân sự
Chia tập train/test cho phép bạn phát triển bộ phân loại trên phần dữ liệu huấn luyện và kiểm tra nó trên phần còn lại của tập dữ liệu. Trong bài tập này, bạn sẽ bắt đầu xây dựng mô hình dự đoán nghỉ việc của nhân viên bằng thuật toán phân loại decision tree. Thuật toán cung cấp phương thức .fit(), dùng để khớp các đặc trưng với mô hình trên tập huấn luyện.
Nhắc lại: cả biến mục tiêu và đặc trưng đã được tách sẵn thành phần train và test (Train: features_train, target_train, Test: features_test, target_test)
Bài tập này là một phần của khóa học
Phân tích Nhân sự (HR Analytics): Dự đoán nghỉ việc bằng Python
Hướng dẫn bài tập
- Import thuật toán phân loại có tên
DecisionTreeClassifier. - Khởi tạo nó thành
modelvà đặt random state là 42. - Áp dụng mô hình decision tree bằng cách fit các đặc trưng của tập huấn luyện vào
model.
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 the classification algorithm
from sklearn.tree import ____
# Initialize it and call model by specifying the random_state parameter
model = ____(random_state=42)
# Apply a decision tree model to fit features to the target
model.____(features_train, ____)