Update 'dvc.yaml'

This commit is contained in:
Kacper Dudzic 2022-05-30 21:10:50 +02:00
parent 5cab54fd20
commit b4a85e2ef3

View File

@ -3,16 +3,8 @@ 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: deps:
- lego_sets_clean_train.csv - lego_sets_clean_train.csv
- lego_sets_clean_test.csv - lego_sets_clean_test.csv
outs:
- lego_reg_model
- lego_reg_results.csv