1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Phân tích Nhân sự (HR Analytics): Dự đoán nghỉ việc bằng Python
Connected

Bài tập

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)

Hướng dẫn

100 XP
  • Import thuật toán phân loại có tên DecisionTreeClassifier.
  • Khởi tạo nó thành model và đặ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.