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.7 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
  1. stages:
  2. data_ingestion:
  3. cmd: python src/deep_classifier/pipeline/stage01_data_ingestion.py
  4. deps:
  5. - src/deep_classifier/pipeline/stage01_data_ingestion.py
  6. - configs/config.yaml
  7. outs:
  8. - src/deep_classifier/artifact/data_ingestion/extracted_data/PetImages
  9. prepare_base_model:
  10. cmd: python src/deep_classifier/pipeline/stage02_prepare_base_model.py
  11. deps:
  12. - src/deep_classifier/pipeline/stage02_prepare_base_model.py
  13. - configs/config.yaml
  14. params:
  15. - params_image_size
  16. - include_top
  17. - classes
  18. - weights
  19. - learning_rate
  20. outs:
  21. - src/deep_classifier/artifact/prepare_base_model/base_model
  22. - src/deep_classifier/artifact/prepare_base_model/updated_model
  23. model_training:
  24. cmd: python src/deep_classifier/pipeline/stage_03_training.py
  25. deps:
  26. - src/deep_classifier/pipeline/stage_03_training.py
  27. - configs/config.yaml
  28. - src/deep_classifier/artifact/prepare_base_model/updated_model
  29. - src/deep_classifier/artifact/data_ingestion/extracted_data/PetImages
  30. params:
  31. - params_image_size
  32. - epochs
  33. - batch_size
  34. - agumentation
  35. outs:
  36. - src/deep_classifier/artifact/prepare_callbacks/model_checkpoint
  37. - src/deep_classifier/artifact/prepare_callbacks/tensorboard_log
  38. model_evaluation:
  39. cmd: python src/deep_classifier/pipeline/stage_04_model_evaluation.py
  40. deps:
  41. - src/deep_classifier/pipeline/stage_04_model_evaluation.py
  42. - configs/config.yaml
  43. - src/deep_classifier/artifact/trained_model/model.h5
  44. - src/deep_classifier/artifact/data_ingestion/extracted_data/PetImages
  45. params:
  46. - params_image_size
  47. - batch_size
  48. outs:
  49. - src/deep_classifier/artifact/trained_model/model.h5
Tip!

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

Comments

Loading...