dvc.yaml 660 B

123456789101112131415161718192021222324252627282930
  1. stages:
  2. featurization:
  3. cmd: python code/featurization.py
  4. deps:
  5. - code/featurization.py
  6. - data/test_data.csv
  7. - data/train_data.csv
  8. outs:
  9. - data/norm_params.json
  10. - data/processed_test_data.npy
  11. - data/processed_train_data.npy
  12. training:
  13. cmd: python code/train_model.py
  14. deps:
  15. - code/train_model.py
  16. - data/processed_train_data.npy
  17. outs:
  18. - data/model.pkl
  19. metrics:
  20. - metrics/train_metric.json:
  21. cache: false
  22. eval:
  23. cmd: python code/eval.py
  24. deps:
  25. - code/eval.py
  26. - data/model.pkl
  27. - data/processed_test_data.npy
  28. metrics:
  29. - metrics/eval.json:
  30. cache: false