undersample_kmeans: Undersample a dataset by kmeans clustering.

In scutr: Balancing Multiclass Datasets for Classification Tasks

Description

Undersample a dataset by kmeans clustering.

Usage

undersample_kmeans(data, cls, cls_col, m, k = 5)

Arguments

data	Dataset to be undersampled.
cls	Class to be undersampled.
cls_col	Column containing class information.
m	Number of samples in undersampled dataset.
k	Number of centers in clustering.

Value

The undersampled dataframe containing only instances of cls.

Examples

table(iris$Species)
undersamp <- undersample_kmeans(iris, "setosa", "Species", 15)
nrow(undersamp)

scutr documentation built on June 24, 2021, 5:07 p.m.