DVC Fixes

2022-06-05 22:19:45 +02:00 · 2022-06-05 22:19:45 +02:00 · 114109ecf6
commit 114109ecf6
parent 3f7b2ec7cf
4 changed files with 9 additions and 5 deletions
--- a/dvc.lock
+++ b/dvc.lock
@ -0,0 +1,4 @@
+schema: '2.0'
+stages:
+  prepare:
+    cmd: download_dataset.sh
--- a/dvc.yaml
+++ b/dvc.yaml
@ -1,6 +1,6 @@
 stages:
  prepare:
-    cmd: ./download_dataset.sh
+    cmd: download_dataset.sh
    
  train:
    cmd: python3 train_model.py
@ -11,4 +11,5 @@ stages:
    plots:
      - trend.png
    params:
-      - model_params.yaml
+      - batch_size
+      - epochs
--- a/model_params.yaml
+++ b/model_params.yaml
@ -1,3 +0,0 @@
-train:
-  batch_size: 64
-  epochs: 5
--- a/params.yaml
+++ b/params.yaml
@ -0,0 +1,2 @@
+batch_size: 64
+epochs: 5