ddi-fw 0.0.176__tar.gz → 0.0.178__tar.gz

{ddi_fw-0.0.176 → ddi_fw-0.0.178}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ddi_fw
-Version: 0.0.176
+Version: 0.0.178
 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 → ddi_fw-0.0.178}/pyproject.toml

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

{ddi_fw-0.0.176 → ddi_fw-0.0.178}/src/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
 
@@ -201,6 +222,12 @@ class BaseDataset(BaseModel):
         self.test_indexes = 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)
 
         if save_indexes:
             # train_pairs = [row['id1'].join(',').row['id2'] for index, row in X_train.iterrows()]

{ddi_fw-0.0.176 → ddi_fw-0.0.178}/src/ddi_fw.egg-info/PKG-INFO

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