2021-06-12 22:59:33 +02:00
|
|
|
stages:
|
|
|
|
|
preprocess:
|
2021-06-12 23:36:06 +02:00
|
|
|
cmd: python3 preprocesing.py 1000
|
2021-06-12 22:59:33 +02:00
|
|
|
deps:
|
|
|
|
|
- preprocesing.py
|
|
|
|
|
- who_suicide_statistics.csv
|
|
|
|
|
outs:
|
|
|
|
|
- train.csv
|
|
|
|
|
train:
|
2021-06-12 23:36:06 +02:00
|
|
|
cmd: python3 training.py 5 10
|
2021-06-12 22:59:33 +02:00
|
|
|
deps:
|
|
|
|
|
- training.py
|
|
|
|
|
- train.csv
|
|
|
|
|
outs:
|
|
|
|
|
- suicide_model.h5
|
|
|
|
|
|
