So sánh các mô hình dự đoán nghỉ việc của nhân viên

Trong bài này, bạn sẽ so sánh mô hình balanced và mô hình imbalanced (mặc định) với cây đã tỉa (max_depth=7). Mô hình imbalanced đã được thực hiện sẵn và đánh giá bằng recall và ROC/AUC. Hãy hoàn thành các bước tương tự cho mô hình balanced.

Các biến features_train, target_train, features_test và target_test đã có sẵn trong không gian làm việc của bạn.
Một mô hình imbalanced đã được fit sẵn cho bạn và dự đoán của nó được lưu trong prediction.
Các hàm recall_score() và roc_auc_score() đã được import sẵn cho bạn.

Khởi tạo mô hình balanced, đặt độ sâu tối đa là 7, và seed là 42.
Fit mô hình vào tập huấn luyện bằng bộ dữ liệu train.
Dự đoán trên bộ dữ liệu test.
In điểm recall và điểm ROC/AUC.