Skip to content

document algorithm in group_vfold_cv #608

Description

@tdhock

I expected some description of the underlying algorithm on ?group_vfold_cv.
In particular I wonder how the assignment works when you specify group, strata, and balance="observations".
Currently I see:

   group: A variable in 'data' (single character or name) used for
          grouping observations with the same value to either the
          analysis or assessment set within a fold.
…
 balance: If 'v' is less than the number of unique groups, how should
          groups be combined into folds? Should be one of '"groups"',
          which will assign roughly the same number of groups to each
          fold, or '"observations"', which will assign roughly the same
          number of observations to each fold.
…
  strata: A variable in 'data' (single character or name) used to
          conduct stratified sampling. When not 'NULL', each resample
          is created within the stratification variable. Numeric
          'strata' are binned into quartiles.

Can you please explain what algorithm is used to assign "roughly the same number of observations to each fold" ?
And can you please clarify, is it

  • roughly the same number of overall observations? or
  • roughly the same number of observations per stratum?

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions