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
|