add_cv_number: Assign a k-fold cross-validation number

In isobelbarrott/Landmarking: Analysis using Landmark Models

Assign a k-fold cross-validation number

Description

Randomly assigns a k-fold cross-validation number to each individual in a dataset.

Usage

add_cv_number(data_long, individual_id, k, seed = 1)

Arguments

data_long	Data frame in long format i.e. there may be more than one row per individual
individual_id	Character string specifying the column name in data_long which contains the individual identifiers
k	Integer specifying the number of folds for cross-validation.
seed	The value of the seed (default is 1)

Details

This function randomly divides the n individual IDs into k groups, each with n/k members (or as close to this number as possible).

Value

Data frame data_long updated to contain a new column cross_validation_number indicating the fold to which the individual has been assigned.

Author(s)

Isobel Barrott [e-mail]

Examples

data(data_repeat_outcomes)
data_repeat_outcomes <- add_cv_number(data_long = data_repeat_outcomes,
                                      individual_id = "id",
                                      k = 10)

isobelbarrott/Landmarking documentation built on Nov. 22, 2022, 4:50 a.m.