parameters: data_partition – R documentation

Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!

data_partition

Partition data into a test and a training set

Description

Creates a training and a test set based on a dataframe. Can also be stratified (i.e., evenly spread a given factor) using the group argument.

Usage

data_partition(x, training_proportion = 0.7, group = NULL, seed = NULL)

Arguments

`x`	A data frame, or an object that can be coerced to a data frame.
`training_proportion`	The proportion (between 0 and 1) of the training set. The remaining part will be used for the test set.
`group`	A character vector indicating the name(s) of the column(s) used for stratified partitioning.
`seed`	A random number generator seed. Enter an integer (e.g., 123) so that the random sampling will be the same each time you run the function.

Value

A list of two data frames, named test and training.

Examples

df <- iris
df$Smell <- rep(c("Strong", "Light"), 75)

head(data_partition(df))
head(data_partition(df, group = "Species"))
head(data_partition(df, group = c("Species", "Smell")))

parameters

Processing of Model Parameters

v0.13.0

GPL-3

Authors

Daniel Lüdecke [aut, cre] (<https://orcid.org/0000-0002-8895-3206>, @strengejacke), Dominique Makowski [aut] (<https://orcid.org/0000-0001-5375-9967>), Mattan S. Ben-Shachar [aut] (<https://orcid.org/0000-0002-4287-4801>), Indrajeet Patil [aut] (<https://orcid.org/0000-0003-1995-6531>, @patilindrajeets), Søren Højsgaard [aut], Zen J. Lau [ctb], Vincent Arel-Bundock [ctb] (<https://orcid.org/0000-0003-1995-6531>, @vincentab), Jeffrey Girard [ctb] (<https://orcid.org/0000-0002-7359-3746>, @jeffreymgirard)

Initial release