ium_464914/dvc.yaml
2024-05-28 21:43:31 +02:00

29 lines
517 B
YAML

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