baby-yoda-segmentor/dvc.yaml

stages:
  split-train-test:
    cmd: poetry run python src/utils.py split-train-test data/raw data/preprocessed
      0.1
    deps:
    - data/raw/annotations
    - data/raw/images
    outs:
    - data/preprocessed/test
    - data/preprocessed/train
  auto-train:
    cmd: python src/train.py auto-train data/preprocessed params.yml models
    deps:
    - data/preprocessed/test
    - data/preprocessed/train
    - params.yml
    - src/train.py
    metrics:
    - metrics.csv:
        cache: false
    outs:
    - models