Spaces:
Running
Running
File size: 2,486 Bytes
f8f5549 |
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 68 69 70 |
from EmotionRecognition import logger
from EmotionRecognition.pipeline.stage_01_data_ingestion import DataIngestionTrainingPipeline
from EmotionRecognition.pipeline.stage_02_data_validation import DataValidationTrainingPipeline
from EmotionRecognition.pipeline.stage_01_data_preparation import DataPreparationPipeline
from EmotionRecognition.pipeline.stage_02_model_training import ModelTrainingPipeline
from EmotionRecognition.pipeline.stage_03_model_evaluation import ModelEvaluationPipeline
# Data Ingestion Stage
STAGE_NAME = "Data Ingestion Stage"
try:
logger.info(f">>>>>> Stage '{STAGE_NAME}' started <<<<<<")
obj = DataIngestionTrainingPipeline()
obj.main()
logger.info(f">>>>>> Stage '{STAGE_NAME}' completed successfully <<<<<<\n\nx==========x")
except Exception as e:
logger.exception(e)
raise e
# Data Validation Stage
STAGE_NAME = "Data Validation Stage"
try:
logger.info(f">>>>>> Stage '{STAGE_NAME}' started <<<<<<")
obj = DataValidationTrainingPipeline()
obj.main()
logger.info(f">>>>>> Stage '{STAGE_NAME}' completed successfully <<<<<<\n\nx==========x")
except Exception as e:
logger.exception(e)
raise e
# Data Preprocessing Stage
#STAGE_NAME = "Data Preprocessing Stage"
#try:
# logger.info(f">>>>>> Stage '{STAGE_NAME}' started <<<<<<")
# obj = DataPreprocessingTrainingPipeline()
# obj.main()
# logger.info(f">>>>>> Stage '{STAGE_NAME}' completed successfully <<<<<<\n\nx==========x")
#except Exception as e:
# logger.exception(e)
# raise e
STAGE_NAME = "Data Preparation Stage"
try:
logger.info(f">>>>>> Stage '{STAGE_NAME}' started <<<<<<")
obj = DataPreparationPipeline()
obj.main()
logger.info(f">>>>>> Stage '{STAGE_NAME}' completed successfully <<<<<<\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 <<<<<<")
obj = ModelTrainingPipeline()
obj.main()
logger.info(f">>>>>> Stage '{STAGE_NAME}' completed successfully <<<<<<\n\nx==========x")
except Exception as e:
logger.exception(e)
raise e
# Model Evaluation Stage
STAGE_NAME = "Model Evaluation Stage"
try:
logger.info(f">>>>>> Stage '{STAGE_NAME}' started <<<<<<")
obj = ModelEvaluationPipeline()
obj.main()
logger.info(f">>>>>> Stage '{STAGE_NAME}' completed successfully <<<<<<\n\nx==========x")
except Exception as e:
logger.exception(e)
raise e |