shad_mlops_transformers/config.py

from pathlib import Path

from pydantic import BaseSettings

basedir = Path(__file__).parent


class Config(BaseSettings):
    data_dir: Path = basedir.parent / "data"
    raw_data_dir: Path = data_dir / "raw"
    checkpoints_folders = data_dir / "checkpoints"
    plots_dir: Path = data_dir / "plots"
    batch_size: int = 64
    epochs: int = 15
    random_seed: int = 42
    test_size: float = 0.2
    weights_path: Path = data_dir / "models" / "model.torch"


config = Config()