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