added dvc.yaml

2023-05-13 12:00:45 +02:00 · 2023-05-13 12:00:45 +02:00 · c4c983d762
commit c4c983d762
parent 0be4bf2ef6
1 changed files with 14 additions and 0 deletions
--- a/dvc.yaml
+++ b/dvc.yaml
@ -0,0 +1,14 @@
+stages:
+    prepare_data:
+        cmd: jupyter nbconvert --to notebook --execute prepare.ipynb
+        deps:
+        - prepare.ipynb
+        outs:
+        - zbior_ium/Car_Prices_Poland_Kaggle.csv
+    train:
+        cmd: python mlflow_train.py
+        deps:
+        - mlflow_train.py
+        - zbior_ium/Cars_Prices_Poland_Kaggle.csv
+        outs:
+        - model.h5