s464961/ium_464914
1
0
Fork 0
forked from s464914/ium_464914
Code Pull Requests Activity

Create dvc.yaml

Browse Source
This commit is contained in:
Alicja Szulecka 2024-05-28 21:43:31 +02:00
parent abb213675e
commit 0f254aa5fa
1 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
dvc.yaml Normal file
View File

@ -0,0 +1,29 @@
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