ddi-fw 0.0.188__tar.gz → 0.0.189__tar.gz

{ddi_fw-0.0.188 → ddi_fw-0.0.189}/PKG-INFO

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

{ddi_fw-0.0.188 → ddi_fw-0.0.189}/pyproject.toml

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

{ddi_fw-0.0.188 → ddi_fw-0.0.189}/src/ddi_fw/datasets/core.py

@@ -158,9 +158,11 @@ class BaseDataset(BaseModel):
         y_test = test[self.class_column]
         
         self.X_train = np.array(X_train)
-        self.y_train = np.array(y_train)
+        # self.y_train = np.array(y_train)
+        self.y_train = np.array(y_train.tolist())
         self.X_test = np.array(X_test)
-        self.y_test = np.array(y_test)
+        # self.y_test = np.array(y_test)
+        self.y_test = np.array(y_test.tolist())
 
         self.train_indexes = X_train.index
         self.test_indexes = X_test.index

{ddi_fw-0.0.188 → ddi_fw-0.0.189}/src/ddi_fw/ml/ml_helper.py

@@ -57,7 +57,7 @@ class MultiModalRunner:
             raise ValueError(
                 "Unsupported library type. Choose 'tensorflow' or 'pytorch'.")
 
-    def __predict(self,single_results):
+    def __predict(self, single_results):
         item_dict = {t[0]: t for t in self.items}
         print("multi_modal")
         print(self.multi_modal)
@@ -71,7 +71,8 @@ class MultiModalRunner:
             model_type = get_import(m.get("model_type"))
             kwargs = m.get('params')
             T = self.__create_model(self.library)
-            single_modal = T(self.date, name, model_type, **kwargs)
+            single_modal = T(self.date, name, model_type,
+                             use_mlflow=self.use_mlflow,  **kwargs)
             if input_type == '1D':
                 item = item_dict[input]
                 single_modal.set_data(
@@ -79,7 +80,7 @@ class MultiModalRunner:
             elif input_type == '2D':
                 # check keys
                 filtered_dict = {k: item_dict[k]
-                                    for k in inputs if k in item_dict}
+                                 for k in inputs if k in item_dict}
                 print(filtered_dict.keys())
                 first_input = next(iter(filtered_dict.values()))
                 train_data_list = [f[1] for f in filtered_dict.values()]

{ddi_fw-0.0.188 → ddi_fw-0.0.189}/src/ddi_fw/ml/tensorflow_wrapper.py

@@ -64,7 +64,7 @@ def convert_to_categorical(arr, num_classes):
 
 class TFModelWrapper(ModelWrapper):
 
-    def __init__(self, date, descriptor, model_func, use_mlflow=True, **kwargs):
+    def __init__(self, date, descriptor, model_func, use_mlflow=False, **kwargs):
         super().__init__(date, descriptor, model_func, **kwargs)
         self.batch_size = kwargs.get('batch_size', 128)
         self.epochs = kwargs.get('epochs', 100)

{ddi_fw-0.0.188 → ddi_fw-0.0.189}/src/ddi_fw.egg-info/PKG-INFO

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