Spaces:
Sleeping
Sleeping
from dataclasses import dataclass | |
from pathlib import Path | |
class DataIngestionConfig: | |
root_dir: Path | |
source_URL: str | |
local_data_file: Path | |
unzip_dir: Path | |
class PrepareBaseModelConfig: | |
root_dir: Path | |
base_model_path: Path | |
updated_base_model_path: Path | |
params_image_size: list | |
params_include_top: bool | |
params_weights: str | |
params_classes: int | |
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_image_size: list | |
class EvaluationConfig: | |
path_of_model: Path | |
training_data: Path | |
all_params: dict | |
mlflow_uri: str | |
params_image_size: list | |
params_batch_size: int | |