stages:
  prepare_data:
    cmd: python ./IUM_2.py
    deps:
      - create-dataset.py
      - covtype.csv
    outs:
      - forest_train.csv
      - forest_test.csv
      - forest_val.csv

  train_model:
    cmd: python ./model.py
    deps:
      - model.py
      - forest_train.csv
      - forest_test.csv
      - forest_val.csv
    outs:
      - model.pth

  evaluate_model:
    cmd: python ./prediction.py
    deps:
      - prediction.py
      - model.pth
      - forest_test.csv
    outs:
      - predictions.txt