Tách Target và Features
Để dự đoán (trong trường hợp này là nhân viên có rời đi hay không), bạn cần tách bộ dữ liệu thành hai phần:
- biến phụ thuộc hay target cần được dự đoán
- các biến độc lập hay features sẽ được dùng để dự đoán
Nhiệm vụ của bạn là tách target và features. Target ở đây là churn của nhân viên, còn features bao gồm tất cả các cột còn lại.
Lưu ý: bộ dữ liệu đã được xử lý sẵn bằng cách mã hóa biến phân loại và tạo biến giả (dummies).
pandas đã được nhập sẵn với bí danh pd.
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
- Thiết lập target và features:
- Chọn cột biến phụ thuộc (
churn) và gán vàotarget. - Dùng
.drop()để bỏ cộtchurnvà lấy phần còn lại làmfeatures.
- Chọn cột biến phụ thuộc (
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.
# Set the target and features
# Choose the dependent variable column (churn) and set it as target
target = data.____
# Drop column churn and set everything else as features
features = data.____("____",axis=1)