dvc.yaml 761 B

123456789101112131415161718192021222324252627282930313233343536
  1. stages:
  2. pre_process:
  3. cmd: python3 src/pre_process.py
  4. deps:
  5. - src/pre_process.py
  6. - data/raw/rML-raw-data.csv
  7. params:
  8. - pre_process
  9. outs:
  10. - data/processed/rML-test.csv
  11. - data/processed/rML-train.csv
  12. train:
  13. cmd: python3 src/train.py
  14. deps:
  15. - src/model_def.py
  16. - data/processed/rML-train.csv
  17. - src/train.py
  18. params:
  19. - train
  20. outs:
  21. - models/model.pkl
  22. metrics:
  23. - models/metrics/train.yaml:
  24. cache: false
  25. eval:
  26. cmd: python3 src/evaluate.py
  27. deps:
  28. - src/evaluate.py
  29. - models/model.pkl
  30. - data/processed/rML-test.csv
  31. params:
  32. - pre_process
  33. - train
  34. metrics:
  35. - models/metrics/test.yaml:
  36. cache: false
Tip!

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