• I'm submitting a ...
  [/] feature request

• Summary
  Implement StartifiedShuffleSplit which is used to sample by assigning a weight to each class.

Reference: https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.StratifiedShuffleSplit.html