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