dvc.yaml

stages:
  prepare:
    cmd: python -m src.features.prepare
    deps:
    - data/raw/test.csv
    - data/raw/train.csv
    - src/features/prepare.py
    params:
    - prepare.random_state
    - prepare.test_size
    - prepare.train_size
    outs:
    - data/processed/X_train.csv
    - data/processed/X_valid.csv
    - data/processed/y_train.csv
    - data/processed/y_valid.csv
  train:
    cmd: python -m src.models.train
    deps:
    - data/processed/X_train.csv
    - data/processed/y_train.csv
    - src/models/train.py
    params:
    - train.algorithm
    - train.random_state
    outs:
    - models/iowa_model.pkl
    metrics:
    - metrics/emissions.csv:
        cache: false
  evaluate:
    cmd: python -m src.models.evaluate
    deps:
    - data/processed/X_valid.csv
    - data/processed/y_valid.csv
    - models/iowa_model.pkl
    - src/models/evaluate.py
    metrics:
    - metrics/scores.json:
        cache: false