-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathapp_.py
26 lines (22 loc) · 1.07 KB
/
app_.py
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
from src.mlproject.logger import logging
from src.mlproject.exception import CustomException
from src.mlproject.components.data_ingestion import DataIngestion
from src.mlproject.components.data_ingestion import DataIngestionConfig
from src.mlproject.components.data_transformation import DataTransformationConfig, DataTransformation
from src.mlproject.components.model_trainer import ModelTrainer
import sys
if __name__ == "__main__":
logging.info("The execution has started")
try:
data_ingestion = DataIngestion()
train_path,test_path = data_ingestion.initiate_data_ingestion()
print("Data Ingestion is completed\n")
data_transformation = DataTransformation()
train_arr,test_arr,_=data_transformation.initiate_data_transformer(train_path,test_path)
print("Data Transformation is completed\n")
# model training
model_trainer = ModelTrainer()
print(model_trainer.initiate_model_trainer(train_arr,test_arr))
except Exception as e:
logging.info("Custom Exception")
raise CustomException(e,sys)