ddi-fw 0.0.225__py3-none-any.whl → 0.0.226__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- ddi_fw/ml/tensorflow_wrapper.py +11 -1
- {ddi_fw-0.0.225.dist-info → ddi_fw-0.0.226.dist-info}/METADATA +1 -1
- {ddi_fw-0.0.225.dist-info → ddi_fw-0.0.226.dist-info}/RECORD +5 -5
- {ddi_fw-0.0.225.dist-info → ddi_fw-0.0.226.dist-info}/WHEEL +0 -0
- {ddi_fw-0.0.225.dist-info → ddi_fw-0.0.226.dist-info}/top_level.txt +0 -0
ddi_fw/ml/tensorflow_wrapper.py
CHANGED
|
@@ -102,9 +102,18 @@ class TFModelWrapper(ModelWrapper):
|
|
|
102
102
|
history = model.fit(
|
|
103
103
|
train_dataset,
|
|
104
104
|
epochs=self.epochs,
|
|
105
|
-
|
|
105
|
+
validation_data=val_dataset,
|
|
106
106
|
callbacks=callbacks
|
|
107
107
|
)
|
|
108
|
+
|
|
109
|
+
# Check if early stopping was applied
|
|
110
|
+
if early_stopping.stopped_epoch > 0:
|
|
111
|
+
print(f"Early stopping was applied at epoch {early_stopping.stopped_epoch}.")
|
|
112
|
+
else:
|
|
113
|
+
print("Early stopping was not applied.")
|
|
114
|
+
if self.tracking_service:
|
|
115
|
+
self.tracking_service.log_param("early_stopping_applied", early_stopping.stopped_epoch > 0)
|
|
116
|
+
self.tracking_service.log_param("early_stopping_epoch", early_stopping.stopped_epoch)
|
|
108
117
|
# ex
|
|
109
118
|
# history = model.fit(
|
|
110
119
|
# X_train, y_train,
|
|
@@ -174,6 +183,7 @@ class TFModelWrapper(ModelWrapper):
|
|
|
174
183
|
if models_val_acc == {}:
|
|
175
184
|
return model, None
|
|
176
185
|
best_model_key = max(models_val_acc, key=lambda k: models_val_acc[k])
|
|
186
|
+
print("best model key: ", best_model_key)
|
|
177
187
|
# best_model_key = max(models_val_acc, key=models_val_acc.get)
|
|
178
188
|
best_model = models[best_model_key]
|
|
179
189
|
return best_model, best_model_key
|
|
@@ -77,7 +77,7 @@ ddi_fw/ml/evaluation_helper.py,sha256=2-7CLSgGTqLEk4HkgCVIOt-GxfLAn6SBozJghAtHb5
|
|
|
77
77
|
ddi_fw/ml/ml_helper.py,sha256=EXMmaSoSmP4RR1zyb1crBE8wwfJohHwWvOhelddtMhI,7945
|
|
78
78
|
ddi_fw/ml/model_wrapper.py,sha256=38uBdHI4H_sjDKPWuhGXovUy_L1tpSNm5tEqCtwmlpY,973
|
|
79
79
|
ddi_fw/ml/pytorch_wrapper.py,sha256=pe6UsjP2XeTgLxDnIUiodoyhJTGCxV27wD4Cjxysu2Q,8553
|
|
80
|
-
ddi_fw/ml/tensorflow_wrapper.py,sha256=
|
|
80
|
+
ddi_fw/ml/tensorflow_wrapper.py,sha256=_mOXMpIkXx7lJySC2wtCDIDhSdtA8bQVEjKwJ5NQ7Io,16782
|
|
81
81
|
ddi_fw/ml/tracking_service.py,sha256=eHWFI3lyQX_xM16CRekgITwldHj2RBMYl5XG8lD8Zks,7508
|
|
82
82
|
ddi_fw/ner/__init__.py,sha256=JwhGXrepomxPSsGsg2b_xPRC72AjvxOIn2CW5Mvscn0,26
|
|
83
83
|
ddi_fw/ner/mmlrestclient.py,sha256=NZta7m2Qm6I_qtVguMZhqtAUjVBmmXn0-TMnsNp0jpg,6859
|
|
@@ -101,7 +101,7 @@ ddi_fw/utils/zip_helper.py,sha256=YRZA4tKZVBJwGQM0_WK6L-y5MoqkKoC-nXuuHK6CU9I,55
|
|
|
101
101
|
ddi_fw/vectorization/__init__.py,sha256=LcJOpLVoLvHPDw9phGFlUQGeNcST_zKV-Oi1Pm5h_nE,110
|
|
102
102
|
ddi_fw/vectorization/feature_vector_generation.py,sha256=EBf-XAiwQwr68az91erEYNegfeqssBR29kVgrliIyac,4765
|
|
103
103
|
ddi_fw/vectorization/idf_helper.py,sha256=_Gd1dtDSLaw8o-o0JugzSKMt9FpeXewTh4wGEaUd4VQ,2571
|
|
104
|
-
ddi_fw-0.0.
|
|
105
|
-
ddi_fw-0.0.
|
|
106
|
-
ddi_fw-0.0.
|
|
107
|
-
ddi_fw-0.0.
|
|
104
|
+
ddi_fw-0.0.226.dist-info/METADATA,sha256=yGP96z5uVKjSep-oOmbxV4FLhJb21yEam1gOVB_HGk0,2631
|
|
105
|
+
ddi_fw-0.0.226.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
|
|
106
|
+
ddi_fw-0.0.226.dist-info/top_level.txt,sha256=PMwHICFZTZtcpzQNPV4UQnfNXYIeLR_Ste-Wfc1h810,7
|
|
107
|
+
ddi_fw-0.0.226.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|