models.group_k_fold()

Grouped k-fold splits: each group’s rows stay in one fold.

Usage

models.group_k_fold(
    groups,
    k,
    seed,
)

Parameters

groups: list of str

k: int

seed: int

Returns

list of (list of int, list of int)