diff --git a/dvc.yaml b/dvc.yaml index 9c05247..3e3c20d 100644 --- a/dvc.yaml +++ b/dvc.yaml @@ -4,16 +4,9 @@ stages: 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 deps: - X_train.csv - train.py - y_train.csv - outs: - - model.pth