ff71d052e6
The training script creates separate model files for each task (match_winner, map_winner, score_team1, score_team2, round_diff, total_maps) so DVC needs to track each file individually. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
38 lines
935 B
YAML
38 lines
935 B
YAML
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
|