18 lines
316 B
YAML
18 lines
316 B
YAML
|
stages:
|
||
|
|
preprocess:
|
||
|
|
cmd: python3 preprocesing.py
|
||
|
|
deps:
|
||
|
|
- preprocesing.py
|
||
|
|
- who_suicide_statistics.csv
|
||
|
|
outs:
|
||
|
|
- train.csv
|
||
|
|
train:
|
||
|
|
cmd: python3 training.py
|
||
|
|
deps:
|
||
|
|
- training.py
|
||
|
|
- train.csv
|
||
|
|
outs:
|
||
|
|
- results.csv
|
||
|
|
- suicide_model.h5
|
||
|
|
|