stages:
  preprocess:
    cmd: python src/data/preprocess.py
    deps:
      - src/data/preprocess.py
      - data/raw
    params:
      - preprocess.test_size
      - preprocess.random_state
    outs:
      - data/processed/features.csv
      - data/processed/train.csv
      - data/processed/test.csv
    metrics:
      - data/processed/data_metrics.json:
          cache: false

  train:
    cmd: python src/models/train_multitask.py
    deps:
      - src/models/train_multitask.py
      - data/processed/train.csv
      - data/processed/test.csv
    params:
      - train.n_estimators
      - train.max_depth
      - train.random_state
    outs:
      - models/model_match_winner.pkl
      - models/model_map_winner.pkl
      - models/model_score_team1.pkl
      - models/model_score_team2.pkl
      - models/model_round_diff.pkl
      - models/model_total_maps.pkl
    metrics:
      - models/metrics.json:
          cache: false