“Comment obtenir des poids de classe tout en utilisant des keras iMagedatagenerator” Réponses codées

Comment obtenir des poids de classe tout en utilisant des keras iMagedatagenerator

from collections import Counter
train_datagen = ImageDataGenerator()
train_generator = train_datagen.flow_from_directory(...)

counter = Counter(train_generator.classes)                          
max_val = float(max(counter.values()))       
class_weights = {class_id : max_val/num_images for class_id, num_images in counter.items()}