21 lines
426 B
YAML
21 lines
426 B
YAML
|
stages:
|
||
|
prepare:
|
||
|
cmd: python3 process_dataset.py
|
||
|
deps:
|
||
|
- lego_sets.csv
|
||
|
- process_dataset.py
|
||
|
outs:
|
||
|
- lego_sets_clean.csv
|
||
|
- lego_sets_clean_train.csv
|
||
|
- lego_sets_clean_valid.csv
|
||
|
- lego_sets_clean_test.csv
|
||
|
train:
|
||
|
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
|
||
|
|