add dvc yaml

2024-05-28 21:50:44 +02:00 · 2024-05-28 21:50:44 +02:00 · 4ade71964d
commit 4ade71964d
parent 563993e596
1 changed files with 17 additions and 0 deletions
--- a/dvc.yaml
+++ b/dvc.yaml
@ -0,0 +1,17 @@
 stages:
  split_data:
    cmd: python get_dataset.py
    deps:
      - Student_Performance.csv
    outs:
      - dataset.csv
      - df_train.csv
      - df_test.csv
  train_model:
    cmd: python train.py 30
    deps:
      - df_train.csv
      - df_test.csv
    outs:
      - model.keras