26 lines
475 B
Python
26 lines
475 B
Python
import numpy as np
|
|
|
|
|
|
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 = np.asarray(y)
|
|
y = np.searchsorted(np.unique(y), y)
|
|
bins = np.bincount(y)
|
|
|
|
weights = 1. / bins.take(y)
|
|
weights *= bins.min()
|
|
|
|
return weights
|