18 lines
320 B
YAML
18 lines
320 B
YAML
|
stages:
|
||
|
|
prepare_data:
|
||
|
|
cmd: python3 clean_and_split_data.py
|
||
|
|
deps:
|
||
|
|
- crime_conv.csv
|
||
|
|
outs:
|
||
|
|
- crime_dev.csv
|
||
|
|
- crime_test.csv
|
||
|
|
- crime_train.csv
|
||
|
|
train:
|
||
|
|
cmd: python3 train_model.py
|
||
|
|
deps:
|
||
|
|
- crime_dev.csv
|
||
|
|
- crime_test.csv
|
||
|
|
- crime_train.csv
|
||
|
|
outs:
|
||
|
|
- model/saved_model.pb
|