Crax Rat File

train_generator = train_datagen.flow_from_directory(train_dir, target_size=(224, 224), batch_size=32, class_mode='categorical')

# Building the model base_model = VGG16(weights='imagenet', include_top=False, input_shape=(224, 224, 3)) crax rat

validation_generator = validation_datagen.flow_from_directory(validation_dir, target_size=(224, 224), batch_size=32, class_mode='categorical') train_generator = train_datagen