|
Download scripts/README.md from alanoee/doodleai: direct link, hf CLI and curl.
- Browser
- Download file 768 Bytes
-
https://huggingface.co/spaces/alanoee/doodleai/resolve/main/scripts/README.md
- Command line
-
hf download hf://spaces/alanoee/doodleai/scripts/README.md
-
curl -L -o README.md https://huggingface.co/spaces/alanoee/doodleai/resolve/main/scripts/README.md
768 Bytes
Scripts
Data preparation and model training scripts.
Files
prepare_data.py- Download and preprocess the Quick Draw! datasettrain_model.py- Train the CNN modelcheck_classes_mapping.py- Verify the number of output classes in a trained model
Usage
# 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.pywrites processed splits todataset/processed/and class mappings tomodel/classes.jsontrain_model.pysaves the best model tomodel/best_model.kerasand training plots tooutputs/