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 1.5 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
  1. stages:
  2. data_ingestion:
  3. cmd: python src/cnnClassifier/pipeline/stage_01_data_ingestion.py
  4. deps:
  5. - src/cnnClassifier/pipeline/stage_01_data_ingestion.py
  6. - config/config.yaml
  7. outs:
  8. - artifacts/data_ingestion/Chest-CT-Scan-data
  9. ## Base Models
  10. prepare_base_model:
  11. cmd: python src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
  12. deps:
  13. - src/cnnClassifier/pipeline/stage_02_prepare_base_model.py
  14. - config/config.yaml
  15. params:
  16. - IMAGE_SIZE
  17. - INCLUDE_TOP
  18. - CLASSES
  19. - WEIGHTS
  20. - LEARNING_RATE
  21. outs:
  22. - artifacts/prepare_base_model
  23. ## train the models and train the trainiing parameters
  24. training:
  25. cmd: python src/cnnClassifier/pipeline/stage_03_model_trainer.py
  26. deps:
  27. - src/cnnClassifier/pipeline/stage_03_model_trainer.py
  28. - config/config.yaml
  29. - artifacts/data_ingestion/Chest-CT-Scan-data
  30. - artifacts/prepare_base_model
  31. params:
  32. - IMAGE_SIZE
  33. - EPOCHS
  34. - BATCH_SIZE
  35. - AUGMENTATION
  36. outs:
  37. - artifacts/training/model.h5
  38. ## Evaluation of models
  39. evaluation:
  40. cmd: python src/cnnClassifier/pipeline/stage_04_model_evaluation.py
  41. deps:
  42. - src/cnnClassifier/pipeline/stage_04_model_evaluation.py
  43. - config/config.yaml
  44. - artifacts/data_ingestion/Chest-CT-Scan-data
  45. - artifacts/training/model.h5
  46. params:
  47. - IMAGE_SIZE
  48. - BATCH_SIZE
  49. metrics:
  50. - scores.json:
  51. cache: false
Tip!

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

Comments

Loading...