run grid search

MoritzM00 · Nov 2, 2024 · ba8a5b5 · ba8a5b5
1 parent c3d18f1
commit ba8a5b5
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/dvc.lock b/dvc.lock
@@ -87,7 +87,7 @@ stages:
       md5: c0f37fa6ec1865b682bb3add8ed17c39
       size: 801854
   train@energy:
-    cmd: uv run python src/probafcst/pipeline/train.py --target energy
+    cmd: python src/probafcst/pipeline/train.py --target energy
     deps:
     - path: data/energy.parquet
       hash: md5
@@ -122,7 +122,7 @@ stages:
       md5: c77ebad39a753f64cd9c62d3ef1f24a3
       size: 2070003
   train@bikes:
-    cmd: uv run python src/probafcst/pipeline/train.py --target bikes
+    cmd: python src/probafcst/pipeline/train.py --target bikes
     deps:
     - path: data/bikes.parquet
       hash: md5

diff --git a/dvc.yaml b/dvc.yaml
@@ -8,7 +8,7 @@ stages:
       - energy
       - bikes
     do:
-      cmd: uv run python ${pipe_dir}/prepare.py --target ${item}
+      cmd: python ${pipe_dir}/prepare.py --target ${item}
       params:
         - data.${item}
         - cache
@@ -22,7 +22,7 @@ stages:
       - energy
       - bikes
     do:
-      cmd: uv run python ${pipe_dir}/train.py --target ${item}
+      cmd: python ${pipe_dir}/train.py --target ${item}
       params:
         - data.${item}
         - train.${item}
@@ -38,7 +38,7 @@ stages:
       - energy
       - bikes
     do:
-      cmd: uv run python ${pipe_dir}/evaluate.py --target ${item}
+      cmd: python ${pipe_dir}/evaluate.py --target ${item}
       params:
         - eval
       deps:
@@ -52,7 +52,7 @@ stages:
       metrics:
         - output/${item}_metrics.json
   submit:
-    cmd: uv run python ${pipe_dir}/submit.py
+    cmd: python ${pipe_dir}/submit.py
     params:
       - quantiles
     deps:

diff --git a/scripts/grid_search.sh b/scripts/grid_search.sh
@@ -11,4 +11,5 @@ for n_weeks in "${n_weeks_values[@]}"; do
             --set-param "train.bikes.benchmark.n_weeks=$n_weeks"
 done
 
-echo "All experiments have been queued. Run 'dvc queue start' to begin processing."
+source .venv/bin/activate
+dvc queue start -v