chore(Machine_Learning): clear notebook outputs and add .gitignore

2026-06-11 16:31:38 +00:00
parent 7a7e5b5955
commit 8c65e33a74
3 changed files with 727 additions and 397 deletions
@@ -25,7 +25,7 @@ def build_clean_model(num_classes, img_size=(224, 224)):
    x = layers.Dense(1024, activation='swish')(x)
    x = layers.BatchNormalization()(x)
    x = layers.Dropout(0.4)(x)
-    outputs = layers.Dense(num_classes, activation='sigmoid', dtype='float32')(x)
+    outputs = layers.Dense(num_classes, activation='softmax', dtype='float32')(x)
    return models.Model(inputs, outputs)

 logging.info("=== EXPORT STARTED ===")