Networksecurity_Project/main.py at main · sufyaansayed78/Networksecurity_Project · GitHub

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
from Networksecurity.Components.data_ingestion import DataIngestion
from Networksecurity.exception.exception import NetwrorkSecurityException
from Networksecurity.Logging.logger import logging
from Networksecurity.entity.config_entity import DataIngestionConfig,DataTransformationConfig,ModelTrainerConfig
from Networksecurity.entity.config_entity import TrainingPipelineConfig
from Networksecurity.Components.data_validation import DataValidation
from Networksecurity.Components.data_validation import DataValidationConfig
from Networksecurity.Components.data_transformation import DataTransformation
from Networksecurity.Components.Model_trainer import ModelTrainer
import sys


if __name__=="__main__":
  try :
    trainingpipelineconfig = TrainingPipelineConfig()
    dataingestionconfig = DataIngestionConfig(trainingpipelineconfig)
    dataingestion = DataIngestion(dataingestionconfig)
    data_validation_config = DataValidationConfig(trainingpipelineconfig)
    logging.info("Initiate the data ingestion")
    dataingestionartifact = dataingestion.initiate_data_ingestion()
    logging.info("Data initiation completed")
    data_validation = DataValidation(dataingestionartifact,data_validation_config)
    logging.info("Initiated Data Validation")
    data_validation_artifact = data_validation.initiate_data_validation()
    logging.info("data validation completed")
    #print(data_validation_artifact)
    data_transformation_config = DataTransformationConfig(training_pipeline_config=trainingpipelineconfig)
    data_transformation = DataTransformation(data_validation_artifact=data_validation_artifact,data_transformation_config=data_transformation_config)
    data_transformation_artifact = data_transformation.initiate_data_transformation()
    #print(data_transformation_artifact)
    logging.info("Data Transformation Completed")
    logging.info("Model Training started")
    model_trainer_config = ModelTrainerConfig(training_pipeline_config=trainingpipelineconfig)
    model_trainer = ModelTrainer(model_trainer_config=model_trainer_config,data_transformation_artifact=data_transformation_artifact)
    model_trainer_artifact=model_trainer.initiate_model_trainer()
    logging.info('Model Trainer artifact')
  except Exception as e :
    raise NetwrorkSecurityException(e,sys)