File size: 768 Bytes
346f830 | 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 | # 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/`
|