ddi-fw 0.0.176__py3-none-any.whl → 0.0.177__py3-none-any.whl

ddi_fw/datasets/core.py

@@ -74,6 +74,21 @@ class BaseDataset(BaseModel):
         arbitrary_types_allowed = True
 
     def produce_inputs(self):
+        items = []
+        if self.X_train is None or self.X_test is None:
+            raise Exception("There is no data to produce inputs")
+        y_train_label, y_test_label = np.array(self.y_train), np.array(self.y_test)
+
+        for index,column in enumerate(self.columns):
+            train_data, test_data =  self.X_train[:,index], self.X_test[:,index]
+            items.append([f'{column}', np.nan_to_num(train_data),
+                          y_train_label, np.nan_to_num(test_data), y_test_label])
+
+            # items.append([f'{column}_embedding', train_data,
+            #             y_train_label, test_data, y_test_label])
+        return items
+    
+    def produce_inputs_ex(self):
         items = []
         if self.X_train is None or self.X_test is None:
             raise Exception("There is no data to produce inputs")
@@ -130,15 +145,21 @@ class BaseDataset(BaseModel):
         train = self.dataframe[self.dataframe.index.isin(train_idx_all)]
         test = self.dataframe[self.dataframe.index.isin(test_idx_all)]
 
-        self.X_train = train.drop(self.class_column, axis=1)
+        self.X_train =  train.drop(self.class_column, axis=1)
         self.y_train = train[self.class_column]
         self.X_test = test.drop(self.class_column, axis=1)
         self.y_test = test[self.class_column]
-
+ 
         self.train_indexes = self.X_train.index
         self.test_indexes = self.X_test.index
         self.train_idx_arr = train_idx_arr
         self.val_idx_arr = val_idx_arr
+        
+        # Dataframe to numpy array conversion
+        self.X_train = np.array(self.X_train)
+        self.y_train = np.array(self.y_train)
+        self.X_test = np.array(self.X_test)
+        self.y_test = np.array(self.y_test)
 
         return self.X_train, self.X_test, self.y_train, self.y_test, self.train_indexes, self.test_indexes, self.train_idx_arr, self.val_idx_arr
 

{ddi_fw-0.0.176.dist-info → ddi_fw-0.0.177.dist-info}/METADATA

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

{ddi_fw-0.0.176.dist-info → ddi_fw-0.0.177.dist-info}/RECORD

@@ -1,5 +1,5 @@
 ddi_fw/datasets/__init__.py,sha256=_I3iDHARwzmg7_EL5XKtB_TgG1yAkLSOVTujLL9Wz9Q,280
-ddi_fw/datasets/core.py,sha256=j6YpH6IqPQ2va1cC26xT-Jn3fIPsF43xD3GuluJRJb4,9372
+ddi_fw/datasets/core.py,sha256=k5rxauXCNWcUW_RHfswB4oTSz-EZyWHboH64o3x8mjg,10328
 ddi_fw/datasets/dataset_splitter.py,sha256=8H8uZTAf8N9LUZeSeHOMawtJFJhnDgUUqFcnl7dquBQ,1672
 ddi_fw/datasets/db_utils.py,sha256=OTsa3d-Iic7z3HmzSQK9UigedRbHDxYChJk0s4GfLnw,6191
 ddi_fw/datasets/setup_._py,sha256=khYVJuW5PlOY_i_A16F3UbSZ6s6o_ljw33Byw3C-A8E,1047
@@ -97,7 +97,7 @@ ddi_fw/utils/zip_helper.py,sha256=YRZA4tKZVBJwGQM0_WK6L-y5MoqkKoC-nXuuHK6CU9I,55
 ddi_fw/vectorization/__init__.py,sha256=LcJOpLVoLvHPDw9phGFlUQGeNcST_zKV-Oi1Pm5h_nE,110
 ddi_fw/vectorization/feature_vector_generation.py,sha256=Z1A_DOBqDFPqLN4YB-3oYlOQWJK-X6Oes6UFjpzR47Q,4760
 ddi_fw/vectorization/idf_helper.py,sha256=_Gd1dtDSLaw8o-o0JugzSKMt9FpeXewTh4wGEaUd4VQ,2571
-ddi_fw-0.0.176.dist-info/METADATA,sha256=PVdIc0cHbS_dAutE06MAWIRWSVlAM7qAtGGN6DLIKCQ,2542
-ddi_fw-0.0.176.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-ddi_fw-0.0.176.dist-info/top_level.txt,sha256=PMwHICFZTZtcpzQNPV4UQnfNXYIeLR_Ste-Wfc1h810,7
-ddi_fw-0.0.176.dist-info/RECORD,,
+ddi_fw-0.0.177.dist-info/METADATA,sha256=LVa6DvyGtARYgDpsqlekfEqeg3nj6uPALn10NKB2eE0,2542
+ddi_fw-0.0.177.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+ddi_fw-0.0.177.dist-info/top_level.txt,sha256=PMwHICFZTZtcpzQNPV4UQnfNXYIeLR_Ste-Wfc1h810,7
+ddi_fw-0.0.177.dist-info/RECORD,,