statgarten / stove

Machine learning modeling, visualization
http://www.statgarten.com/stove/
Other
17 stars 5 forks source link

[FEATURE] Stratified cross validation #23

Open ycseong07 opened 1 year ago

ycseong07 commented 1 year ago

The current implementation of the rsample::vfold_cv() function only includes the 'data' and 'v' parameters. However, to improve a model's performance, it is necessary to include the 'strata' parameter as the target variable.