dvc.yaml 936 B

123456789101112131415161718192021222324252627282930313233343536373839
  1. stages:
  2. train:
  3. cmd: python -m src.train
  4. deps:
  5. - data/raw-data/test
  6. - data/raw-data/train
  7. - src/const/general_const.py
  8. - src/const/train_const.py
  9. - src/train.py
  10. - src/utiles/functions.py
  11. outs:
  12. - eval/processed_data:
  13. persist: true
  14. - eval/metrics/TF_training_logs.csv:
  15. cache: false
  16. persist: true
  17. - model/TF-Model-Checkpoint:
  18. persist: true
  19. eval:
  20. cmd: python -m src.evaluate
  21. deps:
  22. - data/raw-data/val
  23. - model/TF-Model-Checkpoint
  24. - src/const/evaluate_const.py
  25. - src/const/general_const.py
  26. - src/evaluate.py
  27. - src/utiles/functions.py
  28. outs:
  29. - eval/params/TF_train_final_params.yml:
  30. cache: false
  31. persist: true
  32. - eval/predictions:
  33. persist: true
  34. - model/prod_model:
  35. persist: true
  36. metrics:
  37. - eval/metrics/TF_train_final_metrics.csv:
  38. cache: false
  39. persist: true
Tip!

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