EndtoendDS/app.py at main · Devgan79/EndtoendDS · 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
from src.endtoendDS.logger import logging
from src.endtoendDS.exception import CustomException
import sys
from src.endtoendDS.components.data_ingestion import DataIngestion, DataIngestionConfig

from src.endtoendDS.components.data_transformation import DataTransformationConfig, DataTransformation

from src.endtoendDS.components.model_trainer import ModelTrainerConfig, ModelTrainer


if __name__ == "__main__":
    logging.info("The Exceution has Started in app.py")

    try:
        ##### for Data ingestion verfiy ######
        #data_ingestion_config = DataIngestionConfig()
        data_ingestion=DataIngestion()
        #data_ingestion.initiate_data_ingestion() # top 3 is before you write the transormation code when ur connecting it to the db
        train_data_path,test_data_path = data_ingestion.initiate_data_ingestion()

        ##### for Data taransformation verfiy
        #data_transformation_config=DataTransformationConfig()
        data_transformation=DataTransformation()
        #data_transformation.initiate_data_transormation(train_data_path,test_data_path) # top 3 is before you write the model train
        train_arr, test_arr = data_transformation.initiate_data_transormation(train_data_path,test_data_path)

        ###### for model Training  verfiy ########
        model_trainer =ModelTrainer()
        print(model_trainer.initiate_model_trainer(train_arr,test_arr))


    except Exception as e:
        logging.info("CustomException occured")
        raise CustomException(e,sys)