schema: '2.0'
stages:
  train:
    cmd: python3 tf_train.py
    deps:
    - path: stop_times.categories.tsv
      md5: d88f6a43257758c19a8cd8d708459958
      size: 3693
    - path: stop_times.train.tsv
      md5: 4f95b09a81a43d946f97ea05ff83dae0
      size: 59369274
    - path: stop_times.valid.tsv
      md5: 179d6d81c2d284ae84af89c66dcf378d
      size: 133596844
    outs:
    - path: model.keras
      md5: f7136946f9bafe2b6d8a794379292631
      size: 19874552
  test:
    cmd: python3 tf_test.py
    deps:
    - path: model.keras
      md5: f7136946f9bafe2b6d8a794379292631
      size: 19874552
    - path: stop_times.categories.tsv
      md5: d88f6a43257758c19a8cd8d708459958
      size: 3693
    - path: stop_times.test.tsv
      md5: 2b6b49382efb97c511755f21453f8726
      size: 44527766
    outs:
    - path: stop_times.accuracy.tsv
      md5: dd1d9835fa82bf98491de73c74e42ec3
      size: 20
    - path: stop_times.predictions.tsv
      md5: 5e18d5b87fce01e5f87c9939ec84fdca
      size: 16681428