| # Scripts | |
| Data preparation and model training scripts. | |
| ## Files | |
| - `prepare_data.py` - Download and preprocess the Quick Draw! dataset | |
| - `train_model.py` - Train the CNN model | |
| - `check_classes_mapping.py` - Verify the number of output classes in a trained model | |
| ## Usage | |
| ```bash | |
| # 1. Download and preprocess data (~2GB download, configurable sample count) | |
| python scripts/prepare_data.py | |
| # 2. Train the model | |
| python scripts/train_model.py | |
| ``` | |
| ## Configuration | |
| Paths and training parameters are defined in `config.py` at the project root. | |
| ## Output | |
| - `prepare_data.py` writes processed splits to `dataset/processed/` and class mappings to `model/classes.json` | |
| - `train_model.py` saves the best model to `model/best_model.keras` and training plots to `outputs/` | |