createCvTrainTestSets: Create train and test sets for k-fold cross validation

In luannnguyen/mltoolkit: Various tools for training machine learning classifiers

Description

Shuffles observations and splits them into k-fold train/test sets.

Usage

createCvTrainTestSets(
  df,
  k = 10,
  stratify.by.col = NULL,
  return.data = T,
  show.warnings = T
)

Arguments

df	A dataframe of observations as rows and features as columns
k	Number of folds to split the data into
stratify.by.col	(Optional) The name of a column containing the response classes by which to stratify the data. Stratification ensures that all classes are present in every train/test set.
return.data	Return the split dataframe or only indexes
show.warnings	Show warnings?

Value

A list of length(k), containing a sublist of train and test data or indexes

Examples

createCvTrainTestSets(df, k=10, stratify.by.col = 'response')

luannnguyen/mltoolkit documentation built on Aug. 29, 2020, 8:31 a.m.