stages:
  train:
    cmd: python3 tf_train.py
    wdir: src
    deps:
    - stop_times.categories.tsv
    - stop_times.train.tsv
    - stop_times.valid.tsv
    outs:
    - model.keras
  test:
    cmd: python3 tf_test.py
    wdir: src
    deps:
    - stop_times.categories.tsv
    - stop_times.test.tsv
    - model.keras
    outs:
    - stop_times.accuracy.tsv
    - stop_times.predictions.tsv