ddi-fw 0.0.219__py3-none-any.whl → 0.0.221__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/datasets/core.py +1 -1
- ddi_fw/ml/tensorflow_wrapper.py +1 -0
- ddi_fw/pipeline/pipeline.py +1 -1
- {ddi_fw-0.0.219.dist-info → ddi_fw-0.0.221.dist-info}/METADATA +1 -1
- {ddi_fw-0.0.219.dist-info → ddi_fw-0.0.221.dist-info}/RECORD +7 -7
- {ddi_fw-0.0.219.dist-info → ddi_fw-0.0.221.dist-info}/WHEEL +0 -0
- {ddi_fw-0.0.219.dist-info → ddi_fw-0.0.221.dist-info}/top_level.txt +0 -0
ddi_fw/datasets/core.py
CHANGED
@@ -279,7 +279,7 @@ class TextDatasetMixin(BaseModel):
|
|
279
279
|
embedding_dict: Dict[str, Any] | None = Field(
|
280
280
|
default_factory=dict, description="Dictionary for embeddings")
|
281
281
|
pooling_strategy: PoolingStrategy | None = None
|
282
|
-
column_embedding_configs: Optional[
|
282
|
+
column_embedding_configs: Optional[List] = None
|
283
283
|
vector_db_persist_directory: Optional[str] = None
|
284
284
|
vector_db_collection_name: Optional[str] = None
|
285
285
|
_embedding_size: int
|
ddi_fw/ml/tensorflow_wrapper.py
CHANGED
@@ -72,6 +72,7 @@ class TFModelWrapper(ModelWrapper):
|
|
72
72
|
def fit_model(self, X_train, y_train, X_valid, y_valid):
|
73
73
|
self.kwargs['input_shape'] = self.train_data.shape
|
74
74
|
self.num_classes = len(np.unique(y_train, axis=0))
|
75
|
+
self.kwargs['num_classes'] = self.num_classes
|
75
76
|
model = self.model_func(**self.kwargs)
|
76
77
|
checkpoint = ModelCheckpoint(
|
77
78
|
filepath=f'{self.descriptor}_validation.weights.h5',
|
ddi_fw/pipeline/pipeline.py
CHANGED
@@ -22,7 +22,7 @@ class Pipeline(BaseModel):
|
|
22
22
|
dataset_splitter_type: Type[DatasetSplitter] = DatasetSplitter
|
23
23
|
columns: Optional[List[str]] = None
|
24
24
|
embedding_dict: Optional[Dict[str, Any]] = None
|
25
|
-
column_embedding_configs: Optional[
|
25
|
+
column_embedding_configs: Optional[List] = None
|
26
26
|
vector_db_persist_directory: Optional[str] = None
|
27
27
|
vector_db_collection_name: Optional[str] = None
|
28
28
|
embedding_pooling_strategy_type: Type[PoolingStrategy] | None = None
|
@@ -1,5 +1,5 @@
|
|
1
1
|
ddi_fw/datasets/__init__.py,sha256=_I3iDHARwzmg7_EL5XKtB_TgG1yAkLSOVTujLL9Wz9Q,280
|
2
|
-
ddi_fw/datasets/core.py,sha256=
|
2
|
+
ddi_fw/datasets/core.py,sha256=PX6MX4hmeYxIWAKAx7NnJr1fpzR11xA8g8vAjYcQNN8,16936
|
3
3
|
ddi_fw/datasets/dataset_splitter.py,sha256=8H8uZTAf8N9LUZeSeHOMawtJFJhnDgUUqFcnl7dquBQ,1672
|
4
4
|
ddi_fw/datasets/db_utils.py,sha256=xRj28U_uXTRPHcz3yIICczFUHXUPiAOZtAj5BM6kH44,6465
|
5
5
|
ddi_fw/datasets/setup_._py,sha256=khYVJuW5PlOY_i_A16F3UbSZ6s6o_ljw33Byw3C-A8E,1047
|
@@ -77,7 +77,7 @@ ddi_fw/ml/evaluation_helper.py,sha256=2-7CLSgGTqLEk4HkgCVIOt-GxfLAn6SBozJghAtHb5
|
|
77
77
|
ddi_fw/ml/ml_helper.py,sha256=RbFUz4kLkqv6WUXXscDFtYBMWcmv7uWOf8YIR5DBfqA,7701
|
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=8hQitM6r0jVkSi4P5O4qjGYuJFT326JcojCrifVEF_M,16227
|
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
|
@@ -87,7 +87,7 @@ ddi_fw/pipeline/multi_modal_combination_strategy.py,sha256=JSyuP71b1I1yuk0s2ecCJ
|
|
87
87
|
ddi_fw/pipeline/multi_pipeline.py,sha256=npJUXYT31fxD6kpJKSeixjbH5jNfPUwIVG7lRdBszRg,9852
|
88
88
|
ddi_fw/pipeline/multi_pipeline_org.py,sha256=AbErwu05-3YIPnCcXRsj-jxPJG8HG2H7cMZlGjzaYa8,9037
|
89
89
|
ddi_fw/pipeline/ner_pipeline.py,sha256=yp-Met2794EKcgr8_3gqt03l4v2efOdaZuAcIXTubvQ,5780
|
90
|
-
ddi_fw/pipeline/pipeline.py,sha256=
|
90
|
+
ddi_fw/pipeline/pipeline.py,sha256=mCEiFwM80Gs6Uo7QFJNioFHwLTZcwoSVmc52Xos81XY,6133
|
91
91
|
ddi_fw/utils/__init__.py,sha256=WNxkQXk-694roG50D355TGLXstfdWVb_tUyr-PM-8rg,537
|
92
92
|
ddi_fw/utils/categorical_data_encoding_checker.py,sha256=T1X70Rh4atucAuqyUZmz-iFULllY9dY0NRyV9-jTjJ0,3438
|
93
93
|
ddi_fw/utils/enums.py,sha256=19eJ3fX5eRK_xPvkYcukmug144jXPH4X9zQqtsFBj5A,671
|
@@ -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.221.dist-info/METADATA,sha256=k7z-kN4sx19LNp0sT1Lh7RsFCICVrJH939adRiSTfdM,2631
|
105
|
+
ddi_fw-0.0.221.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
|
106
|
+
ddi_fw-0.0.221.dist-info/top_level.txt,sha256=PMwHICFZTZtcpzQNPV4UQnfNXYIeLR_Ste-Wfc1h810,7
|
107
|
+
ddi_fw-0.0.221.dist-info/RECORD,,
|
File without changes
|
File without changes
|