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
64
65
66
67
68
69
70
71
72
  1. stages:
  2. split:
  3. cmd: python src/train_test_split.py -i data/iris.csv -o data/split/
  4. deps:
  5. - data/iris.csv
  6. - src/train_test_split.py
  7. outs:
  8. - data/split
  9. featurize:
  10. cmd: python src/feature_engineering.py -i data/split/ -f data/features/ -o models/pca/
  11. deps:
  12. - data/split
  13. - src/feature_engineering.py
  14. params:
  15. - pca
  16. outs:
  17. - data/features
  18. - models/pca/model.gz
  19. metrics:
  20. - models/pca/metrics.yaml:
  21. cache: false
  22. train_logistic:
  23. cmd: python src/logistic_regression.py -i data/features/ -o models/logistic/
  24. deps:
  25. - data/features
  26. - src/logistic_regression.py
  27. params:
  28. - logistic
  29. outs:
  30. - models/logistic/model.gz
  31. metrics:
  32. - models/logistic/metrics.yaml:
  33. cache: false
  34. train_svc:
  35. cmd: python src/linear_svc.py -i data/features/ -o models/svc/
  36. deps:
  37. - data/features
  38. - src/linear_svc.py
  39. params:
  40. - svc
  41. outs:
  42. - models/svc/model.gz
  43. metrics:
  44. - models/svc/metrics.yaml:
  45. cache: false
  46. train_forrest:
  47. cmd: python src/random_forrest.py -i data/features/ -o models/r_forrest/
  48. deps:
  49. - data/features
  50. - src/random_forrest.py
  51. params:
  52. - forrest
  53. outs:
  54. - models/r_forrest/model.gz
  55. metrics:
  56. - models/r_forrest/metrics.yaml:
  57. cache: false
  58. train_ensemble:
  59. cmd: python src/ensemble.py -i data/features/ -m models/ -o models/ensemble/
  60. deps:
  61. - data/features
  62. - models/logistic/model.gz
  63. - models/r_forrest/model.gz
  64. - models/svc/model.gz
  65. - src/ensemble.py
  66. params:
  67. - ensemble
  68. outs:
  69. - models/ensemble/model.gz
  70. metrics:
  71. - models/ensemble/metrics.yaml:
  72. cache: false
Tip!

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

Comments

Loading...