28 lines
557 B
YAML
28 lines
557 B
YAML
|
stages:
|
||
|
download:
|
||
|
cmd: bash create-dataset.sh 10000
|
||
|
deps:
|
||
|
- create-dataset.sh
|
||
|
outs:
|
||
|
- data/car_prices.csv
|
||
|
- data/car_prices_test.csv
|
||
|
- data/car_prices_dev.csv
|
||
|
- data/car_prices_train.csv
|
||
|
|
||
|
train:
|
||
|
cmd: python model.py
|
||
|
deps:
|
||
|
- model.py
|
||
|
- data/car_prices_train.csv
|
||
|
outs:
|
||
|
- car_prices_predict_model.h5
|
||
|
|
||
|
predict:
|
||
|
cmd: python predict.py
|
||
|
deps:
|
||
|
- predict.py
|
||
|
- data/car_prices_test.csv
|
||
|
- car_prices_predict_model.h5
|
||
|
outs:
|
||
|
- predicted_selling_prices.csv
|