watex.utils.mlutils.stratifiedUsingDiscretedCategories#

watex.utils.mlutils.stratifiedUsingDiscretedCategories(data, cat_name, n_splits=1, test_size=0.2, random_state=42)[source]#

Stratified sampling based on new generated category from DiscretizeCategoriesforStratification().

Parameters:
  • data – dataframe holding the new column of category

  • cat_name – new category name inserted into data

  • n_splits – number of splits