split_data_prob: Create Train Index Set

In stressor: Algorithms for Testing Models under Stress

Create Train Index Set

Description

This function takes in a data.frame object and the training size and returns a logical vector indicating which entries to include.

Usage

split_data_prob(data, test_prop)

Arguments

data	A data.frame object used to determine the length of the vector.
test_prop	A numeric that is between zero and one that represents the proportion of observations to be included in the test data.

Value

A logical vector is returned that is the same length as the number of rows of the data.

Examples

  lm_data <- data_gen_lm(10)
  indices <- split_data_prob(lm_data, .8)
  train <- lm_data[indices, ]
  test <- lm_data[!indices, ]

stressor documentation built on May 29, 2024, 3:09 a.m.