36 lines
700 B
YAML
36 lines
700 B
YAML
stages:
|
|
prepare_data:
|
|
cmd: python ./create-dataset.py
|
|
deps:
|
|
- create-dataset.py
|
|
- creditcard.csv
|
|
outs:
|
|
- data/creditcard.csv
|
|
- data/X_train.csv
|
|
- data/X_val.csv
|
|
- data/X_test.csv
|
|
- data/y_train.csv
|
|
- data/y_val.csv
|
|
- data/y_test.csv
|
|
|
|
train_model:
|
|
cmd: python ./train_model.py
|
|
deps:
|
|
- train_model.py
|
|
- data/X_train.csv
|
|
- data/X_val.csv
|
|
- data/y_train.csv
|
|
- data/y_val.csv
|
|
outs:
|
|
- model/model.keras
|
|
|
|
evaluate_model:
|
|
cmd: python ./predict.py
|
|
deps:
|
|
- predict.py
|
|
- model/model.keras
|
|
- data/X_test.csv
|
|
- data/y_test.csv
|
|
outs:
|
|
- data/y_pred.csv
|