From 73b627f6eeb9f1a484442ea9659aacbc699980f8 Mon Sep 17 00:00:00 2001 From: Marcin Kostrzewski Date: Sun, 5 Jun 2022 21:58:57 +0200 Subject: [PATCH] Add dvc.yaml --- dvc.yaml | 14 ++++++++++++++ model_params.yaml | 3 +++ requirements.txt | 5 ++++- 3 files changed, 21 insertions(+), 1 deletion(-) create mode 100644 dvc.yaml create mode 100644 model_params.yaml diff --git a/dvc.yaml b/dvc.yaml new file mode 100644 index 0000000..f92dd9a --- /dev/null +++ b/dvc.yaml @@ -0,0 +1,14 @@ +stages: + prepare: + cmd: ./download_dataset.sh + + train: + cmd: python3 train_model.py + deps: + - data/Plant_1_Generation_data.csv.train + outs: + - model_out + plots: + - trend.png + params: + - model_params.yaml \ No newline at end of file diff --git a/model_params.yaml b/model_params.yaml new file mode 100644 index 0000000..cbf4e69 --- /dev/null +++ b/model_params.yaml @@ -0,0 +1,3 @@ +train: + batch_size: 64 + epochs: 5 \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 2aa39c9..417886d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,4 +5,7 @@ numpy~=1.22.3 matplotlib==3.5.2 sacred==0.8.2 pymongo==4.1.1 -mlflow==1.25.1 \ No newline at end of file +mlflow==1.25.1 +dvc==2.10.2 +dvc-ssh==0.0.1a0 +paramiko==2.11.0 \ No newline at end of file