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.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
  1. # add stages here
  2. stages:
  3. get_data:
  4. cmd: python src/stage_01_get_data.py --config=configs/config.yaml
  5. deps:
  6. - src/stage_01_get_data.py
  7. - src/utils/common.py
  8. - configs/config.yaml
  9. outs:
  10. - data/data.xml
  11. prepare_data:
  12. cmd: python src/stage_02_prepare.py --config=configs/config.yaml --params=params.yaml
  13. deps:
  14. - src/stage_02_prepare.py
  15. - src/utils/common.py
  16. - src/utils/data_mgmt.py
  17. - data/data.xml
  18. - configs/config.yaml
  19. params:
  20. - prepare.seed
  21. - prepare.split
  22. outs:
  23. - artifacts/prepared/train.tsv
  24. - artifacts/prepared/test.tsv
  25. featurize:
  26. cmd: python src/stage_03_featurization.py --config=configs/config.yaml --params=params.yaml
  27. deps:
  28. - src/stage_03_featurization.py
  29. - src/utils/common.py
  30. - src/utils/featurize.py
  31. - artifacts/prepared/train.tsv
  32. - artifacts/prepared/test.tsv
  33. - configs/config.yaml
  34. params:
  35. - featurize.max_features
  36. - featurize.ngrams
  37. outs:
  38. - artifacts/features/train.pkl
  39. - artifacts/features/test.pkl
  40. train:
  41. cmd: python src/stage_04_train.py --config=configs/config.yaml --params=params.yaml
  42. deps:
  43. - src/stage_04_train.py
  44. - src/utils/common.py
  45. - artifacts/features/train.pkl
  46. - configs/config.yaml
  47. params:
  48. - train.seed
  49. - train.n_est
  50. - train.min_split
  51. outs:
  52. - artifacts/model/model.pkl
  53. evaluate:
  54. cmd: python src/stage_05_evaluate.py --config=configs/config.yaml
  55. deps:
  56. - src/stage_05_evaluate.py
  57. - src/utils/common.py
  58. - artifacts/features/test.pkl
  59. - configs/config.yaml
  60. - artifacts/model/model.pkl
  61. metrics:
  62. - scores.json:
  63. cache: false
  64. plots:
  65. - prc.json:
  66. cache: false
  67. x: recall
  68. y: precision
  69. - roc.json:
  70. cache: false
  71. x: fpr
  72. y: tpr
Tip!

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

Comments

Loading...