Update 'dvc.yaml'

This commit is contained in:
Kacper Dudzic 2022-05-30 21:09:04 +02:00
parent 6a8f4aa868
commit 5cab54fd20

View File

@ -3,6 +3,16 @@ stages:
cmd: python3 process_dataset.py cmd: python3 process_dataset.py
deps: deps:
- lego_sets.csv - lego_sets.csv
outs:
- lego_sets_clean.csv
- lego_sets_clean_train.csv
- lego_sets_clean_valid.csv
- lego_sets_clean_test.csv
train: train:
cmd: python3 simple_regression.py 100 cmd: python3 simple_regression.py 100
deps:
- lego_sets_clean_train.csv
- lego_sets_clean_test.csv
outs:
- lego_reg_model
- lego_reg_results.csv