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

dvc.yaml 3.9 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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
  1. stages:
  2. data_ingestion:
  3. cmd: python src/CreditCard/components/stage_01_data_ingestion.py --config=params.yaml
  4. deps:
  5. - src/CreditCard/components/stage_01_data_ingestion.py
  6. params:
  7. - training_pipeline_config
  8. outs:
  9. - artifact/Stage00_data_ingestion/raw_data/UCI_Credit_Card.csv
  10. - artifact/Stage00_data_ingestion/ingested_data/test.csv
  11. - artifact/Stage00_data_ingestion/ingested_data/train.csv
  12. data_validation:
  13. cmd: python src/CreditCard/components/stage_02_data_validation.py --config=params.yaml --schema_file=./configs/schema.yaml
  14. deps:
  15. - src/CreditCard/components/stage_02_data_validation.py
  16. - configs/schema.yaml
  17. - artifact/Stage00_data_ingestion/ingested_data/test.csv
  18. - artifact/Stage00_data_ingestion/ingested_data/train.csv
  19. outs:
  20. - artifact/stage02_data_validation/validated_test.pkl
  21. - artifact/stage02_data_validation/validated_train.pkl
  22. data_transformation:
  23. cmd: python src/CreditCard/components/stage_03_data_transformation.py --config=params.yaml --schema_file=./configs/schema.yaml --feature_store=./configs/feature_generator.yaml
  24. deps:
  25. - src/CreditCard/components/stage_03_data_transformation.py
  26. - configs/feature_generator.yaml
  27. - artifact/stage02_data_validation/validated_test.pkl
  28. - artifact/stage02_data_validation/validated_train.pkl
  29. outs:
  30. - artifact/stage03_data_transformation/preprocessing/preprocessing_obj.pkl
  31. model_trainer:
  32. cmd: python src/CreditCard/components/stage_04_model_trainer.py --config=params.yaml --schema_file=./configs/schema.yaml --model_config=./configs/model_factory.yaml --feature_store=./configs/feature_generator.yaml
  33. deps:
  34. - src/CreditCard/components/stage_04_model_trainer.py
  35. - configs/model_factory.yaml
  36. - artifact/stage03_data_transformation/preprocessing/preprocessing_obj.pkl
  37. - artifact/stage02_data_validation/validated_train.pkl
  38. params:
  39. - model_trainer_config.models
  40. outs:
  41. - artifact/stage04_model_training/best_model/best_model.pkl:
  42. cache : false
  43. - artifact/stage04_model_training/model_eval_reports/model_params.yaml:
  44. cache: false
  45. metrics:
  46. - artifact/stage04_model_training/model_eval_reports/model_report.html:
  47. cache: false
  48. model_eval:
  49. cmd : python src/CreditCard/components/stage_05_model_eval.py --feature_store=./configs/feature_generator.yaml --config=params.yaml --schema_file=./configs/schema.yaml --model_config=./configs/model_factory.yaml
  50. deps:
  51. - src/CreditCard/components/stage_05_model_eval.py
  52. - artifact/stage04_model_training/best_model/best_model.pkl
  53. - artifact/stage02_data_validation/validated_test.pkl
  54. - artifact/stage02_data_validation/validated_train.pkl
  55. outs:
  56. - artifact/stage_05_model_evaluation/evaluated_model_dir/model_eval_report.html:
  57. cache: false
  58. - artifact/stage_05_model_evaluation/evaluated_model_dir/evaluated_model.pkl:
  59. cache: false
  60. metrics:
  61. - artifact/stage_05_model_evaluation/evaluated_model_dir/model_eval_report.json:
  62. cache: false
  63. model_pusher:
  64. cmd: python src/CreditCard/components/stage_06_model_pusher.py --feature_store=./configs/feature_generator.yaml --config=params.yaml --schema_file=./configs/schema.yaml --model_config=./configs/model_factory.yaml
  65. deps:
  66. - src/CreditCard/components/stage_06_model_pusher.py
  67. - artifact/stage02_data_validation/validated_train.pkl
  68. - artifact/stage02_data_validation/validated_test.pkl
  69. - artifact/stage_05_model_evaluation/evaluated_model_dir/evaluated_model.pkl
  70. outs:
  71. - artifact/stage_06_model_pusher/Model_eval_report/model_eval_report.html:
  72. cache: false
  73. - production_model/best_model.pkl:
  74. cache: false
  75. metrics:
  76. - artifact/stage_06_model_pusher/Model_eval_report/model_eval_report.json:
  77. cache: false
Tip!

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

Comments

Loading...