#!/usr/bin/env python3
# =============================================================================
# Created By : Arthur Thuy
# Created Date: Wed February 28 2024
# =============================================================================
"""Module for constants."""
# =============================================================================
from enum import Enum
METRICS_DICT = {
"NRA": "Non-rejected accuracy",
"CQ": "Classification quality",
"RQ": "Rejection quality",
}
ENTROPY_UNC_LIST = ["TU", "AU", "EU"]
ALL_UNC_LIST = ["TU", "AU", "EU", "confidence"]
GENERAL_UNC_LIST = ["entropy", "confidence"]
UNCERTAINTIES_DICT = {
"TU": "Total uncertainty",
"AU": "Aleatoric uncertainty",
"EU": "Epistemic uncertainty",
"confidence": "Confidence",
}
[docs]
class EntropyUnc(str, Enum):
"""Entropy-based uncertainty types."""
TU = "TU"
AU = "AU"
EU = "EU"
[docs]
class GeneralUnc(str, Enum):
"""General uncertainty types."""
ENTROPY = "entropy"
CONFIDENCE = "confidence"
[docs]
class AllUnc(str, Enum):
"""All uncertainty types."""
TU = "TU"
AU = "AU"
EU = "EU"
CONFIDENCE = "confidence"
[docs]
class Metric(str, Enum):
"""Rejection metrics."""
NRA = "NRA"
CQ = "CQ"
RQ = "RQ"