Spaces:
Sleeping
Sleeping
File size: 1,495 Bytes
4821854 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
stages:
data_ingestion:
cmd: python src/vitClassifier/pipeline/stage_01_data_ingestion.py
deps:
- src/vitClassifier/pipeline/stage_01_data_ingestion.py
- config/config.yaml
outs:
- artifacts/data_ingestion
data_transformation:
cmd: python src/vitClassifier/pipeline/stage_02_data_transformation.py
deps:
- src/vitClassifier/pipeline/stage_02_data_transformation.py
# --- THIS IS THE FIX ---
# Remove the old dependency and add the three new ones.
- artifacts/data_ingestion/train_df.csv
- artifacts/data_ingestion/test_df.csv
- artifacts/data_ingestion/val_df.csv
- config/config.yaml
- params.yaml
outs:
- artifacts/data_transformation
model_training:
cmd: python src/vitClassifier/pipeline/stage_03_model_training.py
deps:
- src/vitClassifier/pipeline/stage_03_model_training.py
- artifacts/data_transformation/train_dataset
- artifacts/data_transformation/val_dataset # Added dependency on val dataset
- config/config.yaml
- params.yaml
outs:
- artifacts/model_training/model
model_evaluation:
cmd: python src/vitClassifier/pipeline/stage_04_model_evaluation.py
deps:
- src/vitClassifier/pipeline/stage_04_model_evaluation.py
- artifacts/data_transformation/test_dataset
- artifacts/model_training/model
- config/config.yaml
metrics:
- artifacts/model_evaluation/metrics.json:
cache: false |