File size: 1,100 Bytes
05f7b3b |
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 52 53 54 |
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 PrepareBaseModelConfig:
root_dir: Path
base_model_path: Path
updated_base_model_path: Path
params_image_size: list
params_learning_rate: float
params_include_top: bool
params_weights: str
params_classes: int
@dataclass(frozen=True)
class PrepareCallbacksConfig:
root_dir : Path
tensorboard_root_log_dir : Path
checkpoint_model_filepath : Path
@dataclass(frozen=True)
class TrainingConfig:
root_dir: Path
trained_model_path: Path
updated_base_model_path: Path
training_data: Path
params_epochs: int
params_batch_size: int
params_is_augmentation: bool
params_image_size: list
@dataclass(frozen=True)
class EvaluationConfig:
path_of_model : Path
training_data : Path
all_params : dict
params_image_size : list
params_batch_size: int |