undersample_hclust: Undersample a dataset by hierarchical clustering.

Description Usage Arguments Value Examples

View source: R/undersample.R

Description

Undersample a dataset by hierarchical clustering.

Usage

1
2
3
4
5
6
7
8
9
undersample_hclust(
  data,
  cls,
  cls_col,
  m,
  k = 5,
  h = NA,
  dist_calc = "euclidean"
)

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.

dist_calc

Distance calculation method. See dist().

Value

Undersampled dataframe containing only cls.

Examples

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

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