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.6 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
  1. stages:
  2. download_kaggle:
  3. cmd: 'python -c "from src.download import download_kaggle; download_kaggle(exec_func=True)" '
  4. params:
  5. - download_kaggle
  6. outs:
  7. - dataset
  8. DataPreprocessor:
  9. cmd: "python -c \"from src.process import DataPreprocessor; DataPreprocessor.load(name='DataPreprocessor').run_and_save()\"\
  10. \ "
  11. deps:
  12. - dataset
  13. params:
  14. - DataPreprocessor
  15. outs:
  16. - nodes/DataPreprocessor/outs.json
  17. MLModel:
  18. cmd: "python -c \"from src.train import MLModel; MLModel.load(name='MLModel').run_and_save()\"\
  19. \ "
  20. deps:
  21. - nodes/DataPreprocessor/outs.json
  22. params:
  23. - MLModel
  24. outs:
  25. - nodes/MLModel/model
  26. metrics:
  27. - nodes/MLModel/metrics_no_cache.json:
  28. cache: false
  29. plots:
  30. - nodes/MLModel/training_history.csv:
  31. cache: false
  32. test_data:
  33. cmd: "python -c \"from src.process import DataPreprocessor; DataPreprocessor.load(name='test_data').run_and_save()\"\
  34. \ "
  35. deps:
  36. - dataset
  37. params:
  38. - test_data
  39. outs:
  40. - nodes/test_data/outs.json
  41. EvaluateModel:
  42. cmd: "python -c \"from src.evaluate import EvaluateModel; EvaluateModel.load(name='EvaluateModel').run_and_save()\"\
  43. \ "
  44. deps:
  45. - nodes/MLModel/metrics_no_cache.json
  46. - nodes/MLModel/model
  47. - nodes/MLModel/training_history.csv
  48. - nodes/test_data/outs.json
  49. metrics:
  50. - nodes/EvaluateModel/metrics_no_cache.json:
  51. cache: false
  52. plots:
  53. - nodes/EvaluateModel/confusion_matrix.csv:
  54. cache: false
  55. template: confusion
  56. x: predicted
  57. y: actual
Tip!

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

Comments

Loading...