daltoolbox (version 1.3.717)

sample_balance: Class balancing (up/down sampling)

Description

Balance class distribution using up-sampling or down-sampling.

Usage

sample_balance(attribute, method = c("down", "up"), seed = NULL)

Value

returns an object of class sample_balance

Arguments

attribute

target class attribute name

method

balancing method: "down" or "up"

seed

optional random seed for reproducibility

Examples

data(iris)
iris_imb <- iris[iris$Species != "setosa", ]
sb <- sample_balance("Species", method = "down", seed = 123)
iris_bal <- transform(sb, iris_imb)
table(iris_bal$Species)