ium_444507/dvc.yaml

16 lines
451 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
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
evaluate:
cmd: python3 lab10_evaluate.py
deps:
- CarPrices_pytorch_model.pkl