ium_444507/dvc.yaml
Adam Wojdyla 61cab884dc
Some checks failed
s444507-evaluation/pipeline/head This commit looks good
s444507-dvc/pipeline/head There was a failure building this commit
s444507-predict-s444356/pipeline/head There was a failure building this commit
444507-training/pipeline/head This commit looks good
dockerfile update
2022-06-05 10:08:50 +02:00

24 lines
670 B
YAML

stages:
prepare:
cmd: python3 script_prepare.py data/Car_Prices_Poland_Kaggle.csv
deps:
- data/Car_Prices_Poland_Kaggle.csv
- script_prepare.py
outs:
- data/Car_Prices_Poland_Kaggle_dev.csv
- data/Car_Prices_Poland_Kaggle_train.csv
- data/Car_Prices_Poland_Kaggle_test.csv
train:
cmd: python3 lab05_deepLearning.py 70
deps:
- data/Car_Prices_Poland_Kaggle_dev.csv
- data/Car_Prices_Poland_Kaggle_train.csv
- data/Car_Prices_Poland_Kaggle_test.csv
outs:
- CarPrices_pytorch_model.pkl
- prediction_results.csv
evaluate:
cmd: python3 lab10_evaluate.py
deps:
- CarPrices_pytorch_model.pkl