ddi-fw 0.0.225__tar.gz → 0.0.226__tar.gz

{ddi_fw-0.0.225 → ddi_fw-0.0.226}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ddi_fw
-Version: 0.0.225
+Version: 0.0.226
 Summary: Do not use :)
 Author-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>
 Maintainer-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>

{ddi_fw-0.0.225 → ddi_fw-0.0.226}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ddi_fw"
-version = "0.0.225"
+version = "0.0.226"
 description = "Do not use :)"
 readme = "README.md"
 authors = [

{ddi_fw-0.0.225 → ddi_fw-0.0.226}/src/ddi_fw/ml/tensorflow_wrapper.py

@@ -102,9 +102,18 @@ class TFModelWrapper(ModelWrapper):
         history = model.fit(
             train_dataset,
             epochs=self.epochs,
-            # validation_data=val_dataset,
+            validation_data=val_dataset,
             callbacks=callbacks
         )
+        
+            # Check if early stopping was applied
+        if early_stopping.stopped_epoch > 0:
+            print(f"Early stopping was applied at epoch {early_stopping.stopped_epoch}.")
+        else:
+            print("Early stopping was not applied.")
+        if self.tracking_service:
+            self.tracking_service.log_param("early_stopping_applied", early_stopping.stopped_epoch > 0)
+            self.tracking_service.log_param("early_stopping_epoch", early_stopping.stopped_epoch)
         # ex
         # history = model.fit(
         #     X_train, y_train,
@@ -174,6 +183,7 @@ class TFModelWrapper(ModelWrapper):
         if models_val_acc == {}:
             return model, None
         best_model_key = max(models_val_acc, key=lambda k: models_val_acc[k])
+        print("best model key: ", best_model_key)
         # best_model_key = max(models_val_acc, key=models_val_acc.get)
         best_model = models[best_model_key]
         return best_model, best_model_key

{ddi_fw-0.0.225 → ddi_fw-0.0.226}/src/ddi_fw.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ddi_fw
-Version: 0.0.225
+Version: 0.0.226
 Summary: Do not use :)
 Author-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>
 Maintainer-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>