21 lines
393 B
YAML
21 lines
393 B
YAML
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
|