Stratified K-fold CV

[anbai106]

Hi team,

I found that when you fit a very imbalanced dataset, the CoxPHFitter models led to a singular error because the k_fold_cross_validation does not ensure the folds include both case & non-case participants.

Do you have any solution for this or are you going to implement Sklearn stratified K-fold CV?

Thanks