From 3820bf2a555bb2fbe0b7779cb44e9fc1e02d7aee Mon Sep 17 00:00:00 2001 From: shivpalSW Date: Mon, 17 Jul 2023 18:04:14 +0530 Subject: [PATCH] updated dvc.yaml file --- dvc.yaml | 67 ++++++++++++++++++++++++++++++-------------------------- 1 file changed, 36 insertions(+), 31 deletions(-) diff --git a/dvc.yaml b/dvc.yaml index 1639b5a..827580a 100644 --- a/dvc.yaml +++ b/dvc.yaml @@ -14,6 +14,7 @@ stages: outs: - artifacts/prepared/train.tsv - artifacts/prepared/test.tsv + reflink: false featurize: cmd: python src/stage_02_featurization.py --config=configs/config.yaml --params=params.yaml @@ -30,41 +31,45 @@ stages: outs: - artifacts/features/train.pkl - artifacts/features/test.pkl + reflink: false train: - cmd: python src/stage_03_train.py --config=configs/config.yaml --params=params.yaml - deps: - - src/stage_03_train.py - - artifacts/features/train.pkl - - src/utils/common.py - - configs/config.yaml - params: - - train.seed - - train.n_est - - train.min_split - outs: - - artifacts/model/model.pkl + cmd: python src/stage_03_train.py --config=configs/config.yaml --params=params.yaml + deps: + - src/stage_03_train.py + - artifacts/features/train.pkl + - src/utils/common.py + - configs/config.yaml + params: + - train.seed + - train.n_est + - train.min_split + outs: + - artifacts/model/model.pkl + reflink: false evaluate: - cmd: python src/stage_04_evaluate.py --config=configs/config.yaml - deps: - - src/stage_04_evaluate.py - - artifacts/features/test.pkl - - src/utils/common.py - - configs/config.yaml - - artifacts/model/model.pkl - metrics: - - scores.json: - cache: false - plots: - - prc.json: - cache: false - x: recall - y: precision - - roc.json: - cache: false - x: fpr - y: tpr + cmd: python src/stage_04_evaluate.py --config=configs/config.yaml + deps: + - src/stage_04_evaluate.py + - artifacts/features/test.pkl + - src/utils/common.py + - configs/config.yaml + - artifacts/model/model.pkl + metrics: + - scores.json: + cache: false + plots: + - prc.json: + cache: false + x: recall + y: precision + - roc.json: + cache: false + x: fpr + y: tpr + + reflink: false run_plots_command: cmd: dvc plots diff \ No newline at end of file