Spaces:
Runtime error
Runtime error
File size: 2,162 Bytes
a0f1e38 b9a1fab 821ffc1 cf8627d 9bd2271 a0f1e38 9bd2271 a0f1e38 b9a1fab 9bd2271 b9a1fab 821ffc1 9bd2271 821ffc1 cf8627d 9bd2271 cf8627d a0f1e38 9bd2271 c426c7f 9bd2271 c426c7f 9bd2271 |
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 55 56 57 58 59 60 61 62 63 64 65 66 67 |
from src.imagecolorization.pipeline.stage01_data_ingestion import DataIngestionPipeline
from src.imagecolorization.pipeline.stage02_data_transformation import DataTransformationPipeline
from src.imagecolorization.pipeline.stage_03_model_building import ModelBuildingPipeline
from src.imagecolorization.pipeline.stage_04_model_trainer import ModelTrainerPipeline
from src.imagecolorization.pipeline.stage_05_model_evaluation import ModelEvaluationPipeLine
from src.imagecolorization.logging import logger
STAGE_NAME = 'Data Ingestion Stage'
try:
logger.info(f">>>>>> stage {STAGE_NAME} started <<<<<<")
data_ingestion = DataIngestionPipeline()
data_ingestion.main()
logger.info(f">>>>>> stage {STAGE_NAME} completed <<<<<<\n\nx==========x")
except Exception as e:
logger.exception(e)
raise e
STAGE_NAME = 'Data Tranasformation Stage'
try:
logger.info(f">>>>>> stage {STAGE_NAME} started <<<<<<")
data_transformation = DataTransformationPipeline()
data_transformation.main()
logger.info(f">>>>>> stage {STAGE_NAME} completed <<<<<<\n\nx==========x")
except Exception as e:
logger.exception(e)
raise e
STAGE_NAME = 'Model Building Stage'
try:
logger.info(f">>>>>> stage {STAGE_NAME} started <<<<<<")
model_building = ModelBuildingPipeline()
model_building.main()
logger.info(f">>>>>> stage {STAGE_NAME} completed <<<<<<\n\nx==========x")
except Exception as e:
logger.exception(e)
raise e
STAGE_NAME = 'Model Training Stage'
try:
logger.info(f">>>>>> stage {STAGE_NAME} started <<<<<<")
model_trianer = ModelTrainerPipeline()
model_trianer.main()
logger.info(f">>>>>> stage {STAGE_NAME} completed <<<<<<\n\nx==========x")
except Exception as e:
logger.exception(e)
raise e
# STAGE_NAME = 'Model Evaluation Stage'
# try:
# logger.info(f">>>>>> stage {STAGE_NAME} started <<<<<<")
# model_trianer = ModelEvaluationPipeLine()
# model_trianer.main()
# logger.info(f">>>>>> stage {STAGE_NAME} completed <<<<<<\n\nx==========x")
# except Exception as e:
# logger.exception(e)
# raise e
|