Stratified K-fold

Như bạn vừa thấy, do chia ngẫu nhiên nên phân phối biến mục tiêu giữa các fold khá khác nhau. Điều này không quá quan trọng với bài thi này, nhưng có thể gây vấn đề trong các bài toán phân loại khi biến mục tiêu bị mất cân bằng mạnh.

Để khắc phục, hãy triển khai chiến lược stratified K-fold với việc phân tầng theo biến mục tiêu. DataFrame train đã có sẵn trong không gian làm việc của bạn.