models.k_fold()

k-fold splits of n rows, shuffled with seed.

Usage

models.k_fold(
    n,
    k,
    seed,
)

Parameters

n: int

k: int

seed: int

Returns

list of (list of int, list of int)

(train, test) row indices per fold.