29 lines
692 B
Plaintext
29 lines
692 B
Plaintext
|
schema: '2.0'
|
||
|
stages:
|
||
|
preprocess:
|
||
|
cmd: python preprocesing.py 1000
|
||
|
deps:
|
||
|
- path: preprocesing.py
|
||
|
md5: 5fc702becd41811c92fc85e4ec899d52
|
||
|
size: 904
|
||
|
- path: who_suicide_statistics.csv
|
||
|
md5: ba0fd3d7333524d077791b70048b5067
|
||
|
size: 1895346
|
||
|
outs:
|
||
|
- path: train.csv
|
||
|
md5: eefeb0d1ce453b32aea98b458002f8b0
|
||
|
size: 164096
|
||
|
train:
|
||
|
cmd: python training.py 5 10
|
||
|
deps:
|
||
|
- path: train.csv
|
||
|
md5: eefeb0d1ce453b32aea98b458002f8b0
|
||
|
size: 164096
|
||
|
- path: training.py
|
||
|
md5: d08d4fc890592243d017e91335c2e1d0
|
||
|
size: 2598
|
||
|
outs:
|
||
|
- path: suicide_model.h5
|
||
|
md5: 9595399d5870d7a0352a7c56c5aaa8f1
|
||
|
size: 12496
|