29 lines
517 B
YAML
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 |