IUM_s464980/dvc.yaml

18 lines
300 B
YAML
Raw Permalink Normal View History

2024-05-28 21:50:44 +02:00
stages:
split_data:
cmd: python get_dataset.py
deps:
- Student_Performance.csv
outs:
- dataset.csv
- df_train.csv
- df_test.csv
train_model:
2024-05-28 21:55:15 +02:00
cmd: python train.py --epochs 30
2024-05-28 21:50:44 +02:00
deps:
- df_train.csv
- df_test.csv
outs:
- model.keras