BalancedAccuracy

Balanced accuracy.

Balanced accuracy is the average of recall obtained on each class. It is used to deal with imbalanced datasets in binary and multi-class classification problems.

Parameters

• cm (river.metrics.confusion.ConfusionMatrix) – defaults to None

  This parameter allows sharing the same confusion matrix between multiple metrics. Sharing a confusion matrix reduces the amount of storage and computation time.

Attributes

• bigger_is_better

  Indicate if a high value is better than a low one or not.

• requires_labels

  Indicates if labels are required, rather than probabilities.

• sample_correction

• works_with_weights

  Indicate whether the model takes into consideration the effect of sample weights

Examples

>>> from river import metrics
>>> y_true = [True, False, True, True, False, True]
>>> y_pred = [True, False, True, True, True, False]

>>> metric = metrics.BalancedAccuracy()
>>> for yt, yp in zip(y_true, y_pred):
...     metric = metric.update(yt, yp)

>>> metric
BalancedAccuracy: 62.50%

>>> y_true = [0, 1, 0, 0, 1, 0]
>>> y_pred = [0, 1, 0, 0, 0, 1]
>>> metric = metrics.BalancedAccuracy()
>>> for yt, yp in zip(y_true, y_pred):
...     metric = metric.update(yt, yp)

>>> metric
BalancedAccuracy: 62.50%

Methods

clone

Return a fresh estimator with the same parameters.

The clone has the same parameters but has not been updated with any data. This works by looking at the parameters from the class signature. Each parameter is either - recursively cloned if it's a River classes. - deep-copied via copy.deepcopy if not. If the calling object is stochastic (i.e. it accepts a seed parameter) and has not been seeded, then the clone will not be idempotent. Indeed, this method's purpose if simply to return a new instance with the same input parameters.

get

Return the current value of the metric.

revert

Revert the metric.

Parameters

• y_true
• y_pred
• sample_weight – defaults to 1.0
• correction – defaults to None

update

Update the metric.

Parameters

• y_true
• y_pred
• sample_weight – defaults to 1.0

works_with

Indicates whether or not a metric can work with a given model.

Parameters

• model (river.base.estimator.Estimator)