17 lines
306 B
YAML
17 lines
306 B
YAML
stages:
|
|
preprocess:
|
|
cmd: python3 preprocesing.py 1000
|
|
deps:
|
|
- preprocesing.py
|
|
- who_suicide_statistics.csv
|
|
outs:
|
|
- train.csv
|
|
train:
|
|
cmd: python3 training.py 5 10
|
|
deps:
|
|
- training.py
|
|
- train.csv
|
|
outs:
|
|
- suicide_model.h5
|
|
|