Skip to content

Commit 5fb9faf

Browse files
committed
Add notebooks for model preparation, training, and evaluation; include data and configuration files
- Created 02_prepare_base_model.ipynb for base model preparation using VGG16. - Created 03_model_trainer.ipynb for training the model with data augmentation options. - Created 04_model_evaluation.ipynb for evaluating the trained model and saving results. - Added data.zip containing necessary datasets for training and evaluation. - Added test.yaml for configuration testing with sample data.
1 parent de4e808 commit 5fb9faf

7 files changed

Lines changed: 1 addition & 1 deletion

File tree

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ Data → Training Pipeline → Model Artifacts → Docker Image
7070
├── model
7171
│ └── model.h5 # Trained CNN model
7272
73-
├── research # Experimental notebooks
73+
├── notebooks # Experimental notebooks
7474
│ ├── 01_data_ingestion.ipynb
7575
│ ├── 02_prepare_base_model.ipynb
7676
│ ├── 03_model_trainer.ipynb
File renamed without changes.
File renamed without changes.

0 commit comments

Comments
 (0)