ddi-fw 0.0.133__py3-none-any.whl → 0.0.135__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/ml_helper.py +1 -1
- ddi_fw/ml/tensorflow_wrapper.py +8 -8
- {ddi_fw-0.0.133.dist-info → ddi_fw-0.0.135.dist-info}/METADATA +1 -1
- {ddi_fw-0.0.133.dist-info → ddi_fw-0.0.135.dist-info}/RECORD +6 -6
- {ddi_fw-0.0.133.dist-info → ddi_fw-0.0.135.dist-info}/WHEEL +0 -0
- {ddi_fw-0.0.133.dist-info → ddi_fw-0.0.135.dist-info}/top_level.txt +0 -0
ddi_fw/ml/ml_helper.py
CHANGED
@@ -33,7 +33,7 @@ class MultiModalRunner:
|
|
33
33
|
# todo model related parameters to config
|
34
34
|
def __init__(self, library, multi_modal):
|
35
35
|
self.library = library
|
36
|
-
self.
|
36
|
+
self.multi_modal = multi_modal
|
37
37
|
self.result = Result()
|
38
38
|
|
39
39
|
def set_data(self, items, train_idx_arr, val_idx_arr, y_test_label):
|
ddi_fw/ml/tensorflow_wrapper.py
CHANGED
@@ -69,27 +69,27 @@ class TFModelWrapper(ModelWrapper):
|
|
69
69
|
models[f'validation_{i}'] = model
|
70
70
|
models_val_acc[f'{self.descriptor}_validation_{i}'] = best_val_acc
|
71
71
|
|
72
|
-
best_model_key = max(models_val_acc, key=models_val_acc.get)
|
73
|
-
best_model = models[best_model_key]
|
74
|
-
return best_model, best_model_key
|
72
|
+
self.best_model_key = max(models_val_acc, key=models_val_acc.get)
|
73
|
+
self.best_model = models[self.best_model_key]
|
74
|
+
# return best_model, best_model_key
|
75
75
|
|
76
76
|
# https://github.com/mlflow/mlflow/blob/master/examples/tensorflow/train.py
|
77
77
|
|
78
|
-
def predict(self
|
79
|
-
pred = best_model.predict(self.test_data)
|
78
|
+
def predict(self):
|
79
|
+
pred = self.best_model.predict(self.test_data)
|
80
80
|
return pred
|
81
81
|
|
82
82
|
def fit_and_evaluate(self):
|
83
83
|
|
84
84
|
with mlflow.start_run(run_name=self.descriptor, description="***", nested=True) as run:
|
85
85
|
print(run.info.artifact_uri)
|
86
|
-
|
87
|
-
pred = self.predict(
|
86
|
+
self.fit()
|
87
|
+
pred = self.predict()
|
88
88
|
logs, metrics = evaluate(
|
89
89
|
actual=self.test_label, pred=pred, info=self.descriptor)
|
90
90
|
metrics.format_float()
|
91
91
|
mlflow.log_metrics(logs)
|
92
|
-
mlflow.log_param('best_cv', best_model_key)
|
92
|
+
mlflow.log_param('best_cv', self.best_model_key)
|
93
93
|
|
94
94
|
utils.compress_and_save_data(
|
95
95
|
metrics.__dict__, run.info.artifact_uri, f'{self.date}_metrics.gzip')
|
@@ -75,10 +75,10 @@ ddi_fw/langchain/sentence_splitter.py,sha256=h_bYElx4Ud1mwDNJfL7mUwvgadwKX3GKlSz
|
|
75
75
|
ddi_fw/langchain/storage.py,sha256=OizKyWm74Js7T6Q9kez-ulUoBGzIMFo4R46h4kjUyIM,11200
|
76
76
|
ddi_fw/ml/__init__.py,sha256=tIxiW0g6q1VsmDYVXR_ovvHQR3SCir8g2bKxx_CrS7s,221
|
77
77
|
ddi_fw/ml/evaluation_helper.py,sha256=o4-w5Xa3t4olLW4ymx_8L-Buhe5wfQEmT2bh4Zz544c,13066
|
78
|
-
ddi_fw/ml/ml_helper.py,sha256=
|
78
|
+
ddi_fw/ml/ml_helper.py,sha256=diDHIzcD0CkdRB3lKu_FFH2frpAUTFfI4xKOAxYA04I,6167
|
79
79
|
ddi_fw/ml/model_wrapper.py,sha256=kc01_TVJuriUvNI6ABnLngnJWvmG_Y7-XJ6XMusLJ8U,1088
|
80
80
|
ddi_fw/ml/pytorch_wrapper.py,sha256=AkG-2sKDXr0IBhgmkbjG0i20OuwQv3mhdvqp6UvJDCA,3716
|
81
|
-
ddi_fw/ml/tensorflow_wrapper.py,sha256
|
81
|
+
ddi_fw/ml/tensorflow_wrapper.py,sha256=-42tZ0W_emPsocxNwBYYpa23rUz-Gr5mXzECV2kF64c,5385
|
82
82
|
ddi_fw/ner/__init__.py,sha256=JwhGXrepomxPSsGsg2b_xPRC72AjvxOIn2CW5Mvscn0,26
|
83
83
|
ddi_fw/ner/mmlrestclient.py,sha256=NZta7m2Qm6I_qtVguMZhqtAUjVBmmXn0-TMnsNp0jpg,6859
|
84
84
|
ddi_fw/ner/ner.py,sha256=BEs9AFljAxOQrC2BEP1raSzRoypcfELS5UTdl4bjTqw,15863
|
@@ -106,7 +106,7 @@ ddi_fw/utils/package_helper.py,sha256=erl8_onmhK-41zQoaED2qyDUV9GQxmT9sdoyRp9_q5
|
|
106
106
|
ddi_fw/utils/py7zr_helper.py,sha256=gOqaFIyJvTjUM-btO2x9AQ69jZOS8PoKN0wetYIckJw,4747
|
107
107
|
ddi_fw/utils/utils.py,sha256=szwnxMTDRrZoeNRyDuf3aCbtzriwtaRk4mHSH3asLdA,4301
|
108
108
|
ddi_fw/utils/zip_helper.py,sha256=YRZA4tKZVBJwGQM0_WK6L-y5MoqkKoC-nXuuHK6CU9I,5567
|
109
|
-
ddi_fw-0.0.
|
110
|
-
ddi_fw-0.0.
|
111
|
-
ddi_fw-0.0.
|
112
|
-
ddi_fw-0.0.
|
109
|
+
ddi_fw-0.0.135.dist-info/METADATA,sha256=dT28lZ8b48augSXTYRKE7HAvlj78HgABqIKeFcGcpu4,1965
|
110
|
+
ddi_fw-0.0.135.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
|
111
|
+
ddi_fw-0.0.135.dist-info/top_level.txt,sha256=PMwHICFZTZtcpzQNPV4UQnfNXYIeLR_Ste-Wfc1h810,7
|
112
|
+
ddi_fw-0.0.135.dist-info/RECORD,,
|
File without changes
|
File without changes
|