95 lines
2.3 KiB
Plaintext
95 lines
2.3 KiB
Plaintext
schema: '2.0'
|
|
stages:
|
|
prepare_data:
|
|
cmd: python ./create-dataset.py
|
|
deps:
|
|
- path: create-dataset.py
|
|
hash: md5
|
|
md5: 0903460139f5b57b9759f4de37b2d5e4
|
|
size: 1531
|
|
- path: creditcard.csv
|
|
hash: md5
|
|
md5: e90efcb83d69faf99fcab8b0255024de
|
|
size: 150828752
|
|
outs:
|
|
- path: data/X_test.csv
|
|
hash: md5
|
|
md5: 46ff52696af9a4c06f6b25639525dda6
|
|
size: 30947960
|
|
- path: data/X_train.csv
|
|
hash: md5
|
|
md5: 7505524c54858300bbd92094092a6c39
|
|
size: 92838653
|
|
- path: data/X_val.csv
|
|
hash: md5
|
|
md5: 4d078882cc1898640ddaf4ad9117f543
|
|
size: 30946540
|
|
- path: data/creditcard.csv
|
|
hash: md5
|
|
md5: 4b81435690147d1e624a8b06c5520629
|
|
size: 155302541
|
|
- path: data/y_test.csv
|
|
hash: md5
|
|
md5: a6bc4827feae19934c4021d1f10f5963
|
|
size: 170893
|
|
- path: data/y_train.csv
|
|
hash: md5
|
|
md5: 8112a5cf4faac882c421bcb7e3d42044
|
|
size: 512656
|
|
- path: data/y_val.csv
|
|
hash: md5
|
|
md5: 1155f648650986d8866eba603b86560c
|
|
size: 170893
|
|
train_model:
|
|
cmd: python ./train_model.py
|
|
deps:
|
|
- path: data/X_train.csv
|
|
hash: md5
|
|
md5: 7505524c54858300bbd92094092a6c39
|
|
size: 92838653
|
|
- path: data/X_val.csv
|
|
hash: md5
|
|
md5: 4d078882cc1898640ddaf4ad9117f543
|
|
size: 30946540
|
|
- path: data/y_train.csv
|
|
hash: md5
|
|
md5: 8112a5cf4faac882c421bcb7e3d42044
|
|
size: 512656
|
|
- path: data/y_val.csv
|
|
hash: md5
|
|
md5: 1155f648650986d8866eba603b86560c
|
|
size: 170893
|
|
- path: train_model.py
|
|
hash: md5
|
|
md5: 00b8bac043f4d7a56dec95f2f1bb1b49
|
|
size: 1540
|
|
outs:
|
|
- path: model/model.keras
|
|
hash: md5
|
|
md5: 1d1df55ad26a8c0689efa4a86a86c217
|
|
size: 1476738
|
|
evaluate_model:
|
|
cmd: python ./predict.py
|
|
deps:
|
|
- path: data/X_test.csv
|
|
hash: md5
|
|
md5: 46ff52696af9a4c06f6b25639525dda6
|
|
size: 30947960
|
|
- path: data/y_test.csv
|
|
hash: md5
|
|
md5: a6bc4827feae19934c4021d1f10f5963
|
|
size: 170893
|
|
- path: model/model.keras
|
|
hash: md5
|
|
md5: 1d1df55ad26a8c0689efa4a86a86c217
|
|
size: 1476738
|
|
- path: predict.py
|
|
hash: md5
|
|
md5: a61388aabf381779b38e2f32a4d0df7b
|
|
size: 660
|
|
outs:
|
|
- path: data/y_pred.csv
|
|
hash: md5
|
|
md5: be150c2fbf1914102b479edbe0a4cf43
|
|
size: 1481012
|