Zaktualizuj 'dvc.yaml'

2022-05-30 21:52:31 +02:00 · 2022-05-30 21:52:31 +02:00 · 4ad41fe072
commit 4ad41fe072
parent 575daa70de
1 changed files with 2 additions and 2 deletions
--- a/dvc.yaml
+++ b/dvc.yaml
@ -1,6 +1,6 @@
 stages:
  prepare_datasets:
-    cmd: python prepare_datasets.py X_train.csv X_test.csv y_train.csv y_test.csv
+    cmd: python prepare_datasets.py
    deps:
    - data/bodyPerformance.csv
    - prepare_datasets.py
@ -10,7 +10,7 @@ stages:
    - y_test.csv
    - y_train.csv
  train:
-    cmd: python train.py model.pth
+    cmd: python train.py
    deps:
    - X_train.csv
    - train.py