Metrics

class stambo.metrics.AP[source]

The Average Precision metric. Defined for Binary classifiers.

class stambo.metrics.BACC[source]

Bases: Metric

The balanced accuracy score.

class stambo.metrics.F1Score[source]

Bases: Metric

The F1 score metric. Defined for Binary classifiers.

class stambo.metrics.MAE[source]

Bases: Metric

The mean absolute error.

class stambo.metrics.MCC[source]

Bases: Metric

The Matthew’s correlation coefficient

class stambo.metrics.MSE[source]

Bases: Metric

The Mean squared error

class stambo.metrics.Metric(metric: Callable, int_input: bool = False)[source]

Bases: object

A wrapper for metrics that take predictions and ground truth labels as two arguments.

class stambo.metrics.QKappa[source]

Bases: Metric

Cohen’s kappa score (quadratic).

class stambo.metrics.ROCAUC[source]

Bases: Metric

The ROC-AUC metric. Defined for Binary classifiers.