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

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

Comments

Loading...