stages:
  train:
    cmd: python3 ./ml_pytorch.py
    deps:
    - understat.csv
  evaluate:
    cmd: python3 ./ml_pytorch_results.py
    deps:
    - understat.csv