eeat-scorer / config.py
Alireza Aminzadeh
Upload folder using huggingface_hub
14beabf verified
raw
history blame contribute delete
564 Bytes
"""
Configuration for EEAT-Scorer.
"""
import os
from pathlib import Path
BASE_DIR = Path(__file__).resolve().parent
DATA_PATH = os.getenv("DATA_PATH", str(BASE_DIR / "data" / "eeat_labels.csv"))
MODEL_DIR = Path(os.getenv("MODEL_DIR", str(BASE_DIR / "models")))
RANDOM_STATE = int(os.getenv("RANDOM_STATE", "42"))
CONTENT_COLUMN = "content"
TARGET_COLUMNS = ["experience", "expertise", "authoritativeness", "trust"]
OPTIONAL_FEATURES = ["referring_domains", "dr"]
try:
MODEL_DIR.mkdir(parents=True, exist_ok=True)
except OSError:
pass