diff --git a/dvc.yaml b/dvc.yaml new file mode 100644 index 0000000..c01374f --- /dev/null +++ b/dvc.yaml @@ -0,0 +1,19 @@ +stages: + train: + cmd: python nn_train.py + deps: + - data/apps_train.csv + - data/apps_test.csv + - data/apps_validate.csv + - nn_train.py + outs: + - results.txt + evaluate: + cmd: python nn_train_eval.py + deps: + - results.txt + - nn_train_eval.py + outs: + - current_results.txt + - metrics.txt + - output.jpg