Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

main.py 2.0 KB

You have to be logged in to leave a comment. Sign In
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
  1. from CarPrediction import logger
  2. from CarPrediction.pipeline.stage01_data_ingestion import DataIngestionTrainingPipeline
  3. from CarPrediction.pipeline.stage02_data_validation import DataValidationTrainingPipeline
  4. from CarPrediction.pipeline.stage03_data_transformation import DataTransformationTrainingPipeline
  5. from CarPrediction.pipeline.stage04_model_trainer import ModelTrainerPipeline
  6. from CarPrediction.pipeline.stage05_model_evaluation import ModelEvaluationPipeline
  7. STAGE_NAME = "Data Ingestion stage"
  8. try:
  9. logger.info(f">>>>> stage {STAGE_NAME} started")
  10. obj = DataIngestionTrainingPipeline()
  11. obj.main()
  12. logger.info(f">>>>> stage {STAGE_NAME} completed <<<<<\n\nx===========x")
  13. except Exception as e:
  14. logger.exception(e)
  15. raise e
  16. STAGE_NAME = "Data Validation stage"
  17. try:
  18. logger.info(f">>>>> stage {STAGE_NAME} started")
  19. obj = DataValidationTrainingPipeline()
  20. obj.main()
  21. logger.info(f">>>>> stage {STAGE_NAME} completed <<<<<\n\nx===========x")
  22. except Exception as e:
  23. logger.exception(e)
  24. raise e
  25. STAGE_NAME = "Data Transformation stage"
  26. try:
  27. logger.info(f">>>>> stage {STAGE_NAME} started")
  28. obj = DataTransformationTrainingPipeline()
  29. obj.main()
  30. logger.info(f">>>>> stage {STAGE_NAME} completed <<<<<\n\nx===========x")
  31. except Exception as e:
  32. logger.exception(e)
  33. raise e
  34. STAGE_NAME = "Model Trainer stage"
  35. try:
  36. logger.info(f">>>>> stage {STAGE_NAME} started")
  37. obj = ModelTrainerPipeline()
  38. obj.main()
  39. logger.info(f">>>>> stage {STAGE_NAME} completed <<<<<\n\nx===========x")
  40. except Exception as e:
  41. logger.exception(e)
  42. raise e
  43. STAGE_NAME = "Model Evaluation stage"
  44. try:
  45. logger.info(f">>>>> stage {STAGE_NAME} started")
  46. obj = ModelEvaluationPipeline()
  47. obj.main()
  48. logger.info(f">>>>> stage {STAGE_NAME} completed <<<<<\n\nx===========x")
  49. except Exception as e:
  50. logger.exception(e)
  51. raise e
Tip!

Press p or to see the previous file or, n or to see the next file

Comments

Loading...