FadQ's picture
Initial HF Space commit (code only)
0bba60c
raw
history blame contribute delete
712 Bytes
"""
Label mapper utilities.
Handles external numeric label outputs that require a label encoder.
Only used for models that output encoded numeric labels (e.g., LightGBM).
"""
def decode_label(encoder, encoded_label: int) -> str:
"""
Decode a numeric label to its string representation using a label encoder.
This is specifically for ML models like LightGBM that output numeric labels
and require an external label encoder for decoding.
Args:
encoder: sklearn LabelEncoder instance
encoded_label: Numeric label from model prediction
Returns:
String label (e.g., 'C3', 'A2', 'P4')
"""
return encoder.inverse_transform([encoded_label])[0]