Spaces:
Running
Running
File size: 597 Bytes
52bb109 57fbf67 85c00b7 91da5b8 85c00b7 52bb109 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import os
from enum import Enum
from pathlib import Path
class ModelSource(str, Enum):
HF = "hf"
LOCAL = "local"
MODEL_SOURCE = ModelSource(os.getenv("MODEL_SOURCE", "hf"))
HF_MODEL = f"cardiffnlp/twitter-roberta-base-sentiment-latest"
DATASET_PATH = Path("data/dataset")
EVAL_SAMPLE_SIZE = int(os.getenv("EVAL_SAMPLE_SIZE", "100"))
EVAL_PERIOD_MIN = float(os.getenv("EVAL_PERIOD_MIN", "1"))
EVAL_BATCH_SIZE = int(os.getenv("EVAL_BATCH_SIZE", "64"))
TRAIN_FRACTION_SIZE = float(os.getenv("TRAIN_FRACTION_SIZE", "0.2"))
EVAL_FRACTION_SIZE = float(os.getenv("EVAL_FRACTION_SIZE", "0.4"))
|