346f830
1
2
3
4
5
6
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)
