[FEATURE] Add the ability to generate training/validation/testing CSV with proportional splits

[sarthakpati]

Is your feature request related to a problem? Please describe.

Currently, the splits getting generated as purely based on k-fold cross validation [ref], which is fine for segmentation/regression/synthesis, but does not give appropriate results for classification.

Describe the solution you'd like

Add the ability to perform stratified k-fold prior to performing training.

Describe alternatives you've considered

N.A.

Additional context

N.A.

[scap3yvt]

Please assign to me.