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
53
54
55
56
57
58
59
60
61
62
63
  1. stages:
  2. data_load:
  3. cmd: python src/stages/data_load.py --config=params.yaml
  4. deps:
  5. - src/stages/data_load.py
  6. - src/utils/data_utils.py
  7. params:
  8. - base
  9. - data_load
  10. outs:
  11. - data/MAGNETIC_TILE_SURFACE_DEFECTS/images
  12. - data/MAGNETIC_TILE_SURFACE_DEFECTS/masks
  13. data_split:
  14. cmd: python src/stages/data_split.py --config=params.yaml
  15. deps:
  16. - src/stages/data_split.py
  17. - src/utils/data_utils.py
  18. - data/MAGNETIC_TILE_SURFACE_DEFECTS/images
  19. - data/MAGNETIC_TILE_SURFACE_DEFECTS/masks
  20. params:
  21. - base
  22. - data_load
  23. - data_split
  24. outs:
  25. - data/MAGNETIC_TILE_SURFACE_DEFECTS/train_images
  26. - data/MAGNETIC_TILE_SURFACE_DEFECTS/train_masks
  27. - data/MAGNETIC_TILE_SURFACE_DEFECTS/test_images
  28. - data/MAGNETIC_TILE_SURFACE_DEFECTS/test_masks
  29. train:
  30. cmd: python src/stages/train.py --config=params.yaml
  31. deps:
  32. - data/MAGNETIC_TILE_SURFACE_DEFECTS/train_images
  33. - data/MAGNETIC_TILE_SURFACE_DEFECTS/train_masks
  34. - src/stages/train.py
  35. - src/utils/train_utils.py
  36. params:
  37. - base
  38. - data_split
  39. - train
  40. outs:
  41. - models/model_pickle_fastai.pkl
  42. metrics:
  43. - training/metrics.json:
  44. cache: false
  45. plots:
  46. - training/plots:
  47. cache: false
  48. evaluate:
  49. cmd: python src/stages/eval.py --config=params.yaml
  50. deps:
  51. - models/model_pickle_fastai.pkl
  52. - data/MAGNETIC_TILE_SURFACE_DEFECTS/test_images
  53. - data/MAGNETIC_TILE_SURFACE_DEFECTS/test_masks
  54. - src/stages/eval.py
  55. - src/utils/eval_utils.py
  56. params:
  57. - base
  58. - train
  59. - data_split
  60. - evaluate
  61. metrics:
  62. - evaluation/metrics.json:
  63. cache: false
Tip!

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

Comments

Loading...