Spaces:
Sleeping
Sleeping
File size: 1,039 Bytes
85bba48 4de2404 cb7cafb f2492e6 fd31bf7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
from dataclasses import dataclass
from pathlib import Path
@dataclass(frozen=True)
class DataIngestionConfig:
root_dir : Path
source_URL : str
local_data_file : Path
unzip_dir : Path
@dataclass(frozen=True)
class DataValidationConfig:
root_dir : Path
STATUS_FILE : str
ALL_REQUIRED_FILES : list
@dataclass(frozen=True)
class DataTransformationConfig:
root_dir : Path
data_path : Path
tokenizer_name : Path
@dataclass(frozen=True)
class ModelTrainerConfig:
root_dir : Path
data_path : Path
model_ckpt : Path
num_train_epochs : int
warmup_steps : int
per_device_train_batch_size : int
weight_decay : float
logging_steps : int
evaluation_strategy: str
eval_steps: int
save_steps: float
gradient_accumulation_steps: int
@dataclass(frozen=True)
class ModelEvaluationConfig:
root_dir : Path
data_path : Path
model_path : Path
all_params: dict
tokenizer_path : Path
metric_file_name : Path
|