customer-churn/main.py at main · PritamDeb68/customer-churn · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from src.pipeline.training_pipeline import TrainingPipeline
from src.pipeline.testing_pipeline import PredictionPipeline
import sys
from src.exception_handler import CustomException

def invoke_training_pipeline():
    try:
        training_pipeline=TrainingPipeline()
        training_pipeline.initiate_training_pipeline()

    except Exception as e:
        raise  CustomException(e,sys)
def invoke_testing_pipeline():
    try:
        pass
        # testing_pipeline=PredictionPipeline(Age)
        # testing_pipeline.initiate_training_pipeline()

    except Exception as e:
        raise  CustomException(e,sys)

if __name__=="__main__":
    invoke_training_pipeline()