Spaces:

MicroPyscho
/

AidRenal

Sleeping

AidRenal / src /cnnClassifier /entity /config_entity.py

Okereke Kelechi Collins

running 2nd model

36b8bfd 5 months ago

1.06 kB

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
    params_model_name: str
    #source_URL: str
    #local_data_file: Path
    #unzip_dir: 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
    all_params: dict
    
    
@dataclass(frozen=True)
class EvaluationConfig:
    path_of_model: Path
    training_data: Path
    all_params: dict
    mlflow_uri: str
    params_image_size: list
    params_batch_size: int

	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
	params_model_name: str
	#source_URL: str
	#local_data_file: Path
	#unzip_dir: 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
	all_params: dict


	@dataclass(frozen=True)
	class EvaluationConfig:
	path_of_model: Path
	training_data: Path
	all_params: dict
	mlflow_uri: str
	params_image_size: list
	params_batch_size: int