model_diagnosis / service /src /config.py
Volodimirich's picture
Upload folder using huggingface_hub
5cd3fef verified
Raw
History Blame Contribute Delete
968 Bytes
from enum import StrEnum
class Keys(StrEnum):
DIAG = "diag"
DOC = "doc"
class Metrics(StrEnum):
PR = "precision"
RE = "recall"
F1 = "f1"
class ScoreNames(StrEnum):
OPT = "optimistic"
AVG = "averaged"
REAL = "realistic"
class Models(StrEnum):
GIGA_MAX = "giga_max"
GIGA_PLUS = "giga_plus"
GIGA_PRO = "giga_pro"
qwen = "qwen"
deepseekr1 = "deepseekr1"
deepseekv3 = "deepseekv3"
llama = "llama_405b"
mistral = "mistral"
gpt_4o = "gpt4o"
deepseekr1distqwen32b = "deepseekr1distqwen32b"
class Paths:
CHATS = "../data/input/chats_1-360.json"
PREDICTS = "../data/input/predicts_1-360"
TARGETS = "../data/input/targets_1-360.json"
PAIRS = "../data/input/pair-match.json"
ONE_VS_ONE = "../data/input/one-vs-one"
METRICS = "../data/output/metrics_1-360.json"
PAIR_METRICS = "../data/output/pair_metrics_1-360.json"
PREPROCESSOR = "../data/input/preprocessor.json"