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
|