titanic/dvc.yaml

stages:
  feature:
    cmd: python src/kitchen_sink/featurize.py data/data data/feature
    deps:
    - src/kitchen_sink/featurize.py
    - data/data
    params:
    - ft.n_neigh
    outs:
    - data/feature:
        persist: true
  train:
    cmd: python src/kitchen_sink/train.py data/feature data/model score.json prc.json
    deps:
    - data/feature
    - src/kitchen_sink/train.py
    params:
    - train.split
    outs:
    - data/model:
        persist: true
    metrics:
    - score.json:
        cache: false
    plots:
    - prc.json:
        cache: false

        x: recall
        y: precision