Download scripts/check_classes_mapping.py from alanoee/doodleai: direct link, hf CLI and curl.
- Browser
- Download file 173 Bytes
-
https://huggingface.co/spaces/alanoee/doodleai/resolve/main/scripts/check_classes_mapping.py
- Command line
-
hf download hf://spaces/alanoee/doodleai/scripts/check_classes_mapping.py
-
curl -L -o check_classes_mapping.py https://huggingface.co/spaces/alanoee/doodleai/resolve/main/scripts/check_classes_mapping.py
173 Bytes
| import tensorflow as tf | |
| model = tf.keras.models.load_model('model/best_model.keras') | |
| output_layer = model.layers[-1] | |
| print("Number of output classes:", output_layer.units) | |