config.py

"""Constantes compartilhadas pelo Space (bge-m3 FT-Solo, single-fold calibrado)."""
from __future__ import annotations

import os
from pathlib import Path

MODEL_NAME = "BAAI/bge-m3"

MAX_LENGTH = 256
BATCH_SIZE = 8
TASK_PROMPT = None

ROOT         = Path(__file__).resolve().parent
ARTIFACTS_DIR = ROOT / "artifacts"
ADAPTER_PATH  = ARTIFACTS_DIR / "fold_04_adapter"
HEAD_PATH     = ARTIFACTS_DIR / "fold_04_head.pt"

THRESHOLD_UTIL        = 0.5
CONFIDENCE_BOUNDS_ALTA  = (0.10, 0.90)
CONFIDENCE_BOUNDS_MEDIA = (0.30, 0.70)

# Platt scaling: P_calib = sigmoid(CALIB_A * logit(P_raw) + CALIB_B)
# Com defaults (1.0, 0.0) = identidade (sem ajuste).
# Para calibrar: ajuste esses valores com um conjunto de validação.
CALIB_A = 1.0
CALIB_B = 0.0

HF_TOKEN = os.environ.get("HF_TOKEN")