scikit-learn/sklearn/preprocessing/_weights.py at debian · IntelPython/scikit-learn

This repository was archived by the owner on Jan 25, 2023. It is now read-only.

34 lines (25 loc) · 709 Bytes

import numpy as np

from sklearn.utils import safe_asarray

from sklearn.utils import deprecated

def _balance_weights(y):

"""Compute sample weights such that the class distribution of y becomes

balanced.

Parameters

----------

y : array-like

Labels for the samples.

Returns

-------

weights : array-like

The sample weights.

"""

y = safe_asarray(y)

y = np.searchsorted(np.unique(y), y)

bins = np.bincount(y)

weights = 1. / bins.take(y)

weights *= bins.min()

return weights

@deprecated('balance_weights is an internal function and will be removed '

'in 0.16')

def balance_weights(y):

return _balance_weights(y)

Provide feedback