generate_kfold_indices: Generate K-fold Indices for Cross-Validation

In precmed: Precision Medicine

Generate K-fold Indices for Cross-Validation

Description

This function generates indices for K-fold cross-validation based on the total sample size 'N' and the number of folds 'Kfold'. If 'reverse = TRUE', the remainder indices will be assigned in reverse order.

Usage

generate_kfold_indices(N, Kfold, reverse = FALSE)

Arguments

N	Integer. Total sample size (number of observations).
Kfold	Integer. The number of folds to split the data into.
reverse	Logical. Whether to reverse the remainder indices when 'N' is not divisible by 'Kfold'. Defaults to 'FALSE'.

Value

A vector of length 'N' containing the fold assignments (from 1 to 'Kfold').

Author(s)

Thomas Debray

precmed documentation built on Oct. 6, 2024, 1:07 a.m.