Spaces:
Sleeping
Sleeping
File size: 608 Bytes
26d66c1 |
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 |
# -*- coding: utf-8 -*-
import os
from pathlib import Path
from src.logging import logger
from src.models.pipeline.stage_01_model_training import ModelTrainingPipeline
def main():
""" Run model training pipeline """
try:
logger.info('>>>>> Model Training pipeline started <<<<<')
model_training_pipeline = ModelTrainingPipeline()
model_training_pipeline.main()
logger.info('>>>>> Model Training pipeline completed <<<<<')
except Exception as e:
logger.exception(e)
raise e
if __name__ == '__main__':
os.chdir("../../")
main()
|