kangaroo-detection/params.yaml

# Ref: Yaml variable: https://stackoverflow.com/questions/4150782/use-yaml-with-variables
# ---- Variables ----
data:
  main-data: &main-data
    path: 'data/kangaroo'
    size: 164
    train_size: 150
  sample-data: &sample-data
    path: 'data/sample'
    size: 10
    train_size: 5
  test-data: &test-data
    path: 'data/kangaroo-kaggle'
    size: 313
    train_size: NULL

# ---- Misc ----
src:
  pipeline:
    environment-check: 'src/pipeline/environment_check.py'
    train-model: 'src/pipeline/train_model.py'
    evaluate: 'src/pipeline/evaluate.py'

# ---- Stages/Pipelines ----
gen-dag:
  output: out/dependency_diagram.md

environment-check:
  environment-name: kangaroo
  output: out/environment-check.out.json

train-pipeline:
  data: &training-data 
    <<: *sample-data 
    # <<: *main-data
  train_dir: 'model/training'
  output: 
    model: &training-weight
      'out/current-model.h5'
    log: 'out/model-train.output.log'
    metric: 
      folder: out/train.metrics/
      general: 'general.json'
      others: 
        - epoch_mrcnn_bbox_loss
        - epoch_val_mrcnn_bbox_loss

  summary: 'out/model-train.summary.txt'

evaluate:
  data: *training-data
  model-weight: *training-weight
  evaluate_dir: 'model/evaluate'
  output: 
    metric: 'out/evaluate.json'
    log: 'out/evaluate.output.log'