undersample_hclust: Undersample a dataset by hierarchical clustering.

In scutr: Balancing Multiclass Datasets for Classification Tasks

Undersample a dataset by hierarchical clustering.

Description

Undersample a dataset by hierarchical clustering.

Usage

undersample_hclust(data, cls, cls_col, m, k = 5, h = NA, ...)

Arguments

data	Dataset to be undersampled.
cls	Majority class that will be undersampled.
cls_col	Column in data containing class memberships.
m	Number of samples in undersampled dataset.
k	Number of clusters to derive from clustering.
h	Height at which to cut the clustering tree. k must be NA for this to be used.
...	Additional arguments passed to dist().

Value

Undersampled dataframe containing only cls.

Examples

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

scutr documentation built on Nov. 18, 2023, 1:08 a.m.