This module contains the:

  • MetricEnum: The main end user enumeration for metrics.

To build a MetricEvaluator from a configuration, you have to use the Enumerator values as keys.

MetricEnum

Bases: Enum

Enumeration of available metrics.

Source code in metrics_toolbox/metrics/registry.py 
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
class MetricEnum(Enum):
    """Enumeration of available metrics."""

    ACCURACY = Accuracy

    F1_SCORE_MICRO = F1ScoreMicro
    F1_SCORE_MACRO = F1ScoreMacro
    F1_SCORE_TARGET = F1ScoreTarget

    PRECISION_MICRO = PrecisionMicro
    PRECISION_MACRO = PrecisionMacro
    PRECISION_TARGET = PrecisionTarget

    RECALL_MICRO = RecallMicro
    RECALL_MACRO = RecallMacro
    RECALL_TARGET = RecallTarget

    ROC_AUC_MICRO = RocAucMicro
    ROC_AUC_MACRO = RocAucMacro
    ROC_AUC_TARGET = RocAucTarget

    MSE_TARGET = MSETarget
    MSE_MACRO = MSEMacro
    RMSE_TARGET = RMSETarget
    RMSE_MACRO = RMSEMacro