18 lines
300 B
YAML
18 lines
300 B
YAML
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 --epochs 30
|
|
deps:
|
|
- df_train.csv
|
|
- df_test.csv
|
|
outs:
|
|
- model.keras
|