20 lines
388 B
YAML
20 lines
388 B
YAML
stages:
|
|
prepare_datasets:
|
|
cmd: python prepare_datasets.py X_train.csv X_test.csv y_train.csv y_test.csv
|
|
deps:
|
|
- data/bodyPerformance.csv
|
|
- prepare_datasets.py
|
|
outs:
|
|
- X_test.csv
|
|
- X_train.csv
|
|
- y_test.csv
|
|
- y_train.csv
|
|
train:
|
|
cmd: python train.py model.pth
|
|
deps:
|
|
- X_train.csv
|
|
- train.py
|
|
- y_train.csv
|
|
outs:
|
|
- model.pth
|