MKPatel007/network-security-system

1import sys2from networksecurity.components.data_ingestion import DataIngestion3from networksecurity.components.data_validation import DataValidation4from networksecurity.components.model_trainer import ModelTrainer5from networksecurity.exception.exception import NetworkSecurityException6from networksecurity.logging.logger import logging7from networksecurity.entity.config_entity import DataIngestionConfig, DataValidationConfig, ModelTrainerConfig8from networksecurity.entity.config_entity import TrainingPipelineConfig9from networksecurity.components.data_transformation import DataTransformation,DataTransformationConfig10 11if __name__ == '__main__':12    try:13        trainingpipelineconfig = TrainingPipelineConfig()14        dataingestionconfig = DataIngestionConfig(trainingpipelineconfig)15        data_ingestion = DataIngestion(dataingestionconfig)16        logging.info("Initialte the data ingestion")17        18        dataingestionartifacts = data_ingestion.initiate_data_ingestion()19        logging.info("Data initiate completed")20        print(dataingestionartifacts)21        22        data_validation_confing = DataValidationConfig(training_pipeline_config=trainingpipelineconfig)23        data_validation = DataValidation(data_ingestion_artifacts=dataingestionartifacts,data_validatio_config=data_validation_confing)24        logging.info('Initiate data validation')25        data_validation_artifacts = data_validation.initiate_data_validation()26        logging.info("Data validation completed")27        print(data_validation_artifacts)28        29        data_transformation_config = DataTransformationConfig(30            training_pipeline_config=trainingpipelineconfig31        )32        data_transformation = DataTransformation(33            data_validation_artifacts=data_validation_artifacts,34            data_transformation_config=data_transformation_config35        )36        logging.info("Initiate data transformation")37        data_transformation_artifacts = data_transformation.initiate_data_transformation()38        logging.info("Data transformation completed")39        print(data_transformation_artifacts)40        41        logging.info("Model training started")42        model_trainer_config = ModelTrainerConfig(trainingpipelineconfig)43        model_trainer = ModelTrainer(44            model_trainer_config=model_trainer_config,45            data_transformation_artifact=data_transformation_artifacts46        )47        model_trainer_artifact = model_trainer.initiate_model_trainer()48        logging.info("Model Training artifact created")49        50    except Exception as e:51        raise NetworkSecurityException(e,sys)