Cookiecutter-MLOps/dvc.yaml

stages:
  process_data:
    cmd: python -m src.data.make_dataset "data/raw/movie_review.csv" "data/processed/"
    deps:
      - data/raw
      - src/data/make_dataset.py
    outs:
      - data/processed:
          persist: true
  train:
    cmd: python -m src.models.train_model "data/processed/" "models/"
    deps:
    - data/processed
    - src/models/train_model.py
    outs:
    - models:
        persist: true
    - src/models/training_params.yml:
        cache: false
    metrics:
    - reports/training_metrics.csv:
        cache: false
  eval:
    cmd: python -m src.models.predict_model "data/processed/" "models/"
    deps:
    - data/processed
    - models
    - src/models/predict_model.py
    metrics:
    - reports/metrics.csv:
        cache: false