scitex_ml.sampling.undersample

Functions

undersample(X, y[, random_state])

Undersample data preserving input type.

scitex_ml.sampling.undersample.undersample(X, y, random_state=42)

Undersample data preserving input type.

Parameters:

• X (Union[List, Tuple, ndarray, Series, DataFrame, DataArray, Tensor]) – Features array-like of shape (n_samples, n_features)

• y (Union[List, Tuple, ndarray, Series, DataFrame, DataArray, Tensor]) – Labels array-like of shape (n_samples,)

Return type:

Tuple[Union[List, Tuple, ndarray, Series, DataFrame, DataArray, Tensor], Union[List, Tuple, ndarray, Series, DataFrame, DataArray, Tensor]]

Returns:

Resampled X, y of same type as input

Raises:

ImportError – If imblearn is not installed