Waseem2106/network-security

1from networkSecurity.components.data_ingestion import DataIngestion
2from networkSecurity.components.data_validation import DataValidation
3from networkSecurity.exception.exception import NetworkSecurityException
4from networkSecurity.logging.logger import logging
5from networkSecurity.entity.config_entity import DataIngestionConfig,DataValidationConfig
6from networkSecurity.entity.config_entity import TrainingPipelineConfig
7from networkSecurity.components.data_transformation import DataTransformation,DataTransformationConfig
8from networkSecurity.components.model_trainer import ModelTrainerConfig,ModelTrainer
9import sys
10
11if __name__=='__main__':
12    try:
13        trainingPipelineConfig=TrainingPipelineConfig()
14        dataingestionconfig=DataIngestionConfig(trainingPipelineConfig)
15        data_ingestion=DataIngestion(dataingestionconfig)
16        logging.info("Initiate the data ingestion")
17        dataingestionartifact=data_ingestion.initiate_data_ingestion()
18
19        print(dataingestionartifact)
20        logging.info("Data initiation completed")
21
22        data_validation_config=DataValidationConfig(trainingPipelineConfig)
23        data_validation=DataValidation(dataingestionartifact,data_validation_config)
24        logging.info("Initiate the data validation")
25        data_validation_artifact=data_validation.initiate_data_validation()
26        logging.info("data validation completed")
27        print(data_validation_artifact)
28
29        data_transformation_config=DataTransformationConfig(trainingPipelineConfig)
30        logging.info("Data Transformation started")
31        data_transformation=DataTransformation(data_validation_artifact,data_transformation_config)
32        data_transformation_artifact=data_transformation.initiate_data_transformation()
33        print(data_transformation_artifact)
34        logging.info("Data Transformation completed")
35
36        logging.info("Model Trainer started")
37        model_trainer_config=ModelTrainerConfig(trainingPipelineConfig)
38        model_trainer=ModelTrainer(model_trainer_config=model_trainer_config,data_transformation_artifact=data_transformation_artifact)
39        model_trainer_artifact=model_trainer.initiate_model_trainer()
40        logging.info("Model Trainer artifact created")
41    except Exception as e:
42        raise NetworkSecurityException(e,sys)