dragon-ml-toolbox 20.1.1__tar.gz → 20.2.0__tar.gz

{dragon_ml_toolbox-20.1.1/dragon_ml_toolbox.egg-info → dragon_ml_toolbox-20.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 20.1.1
+Version: 20.2.0
 Summary: Complete pipelines and helper tools for data science and machine learning projects.
 Author-email: Karl Luigi Loza Vidaurre <luigiloza@gmail.com>
 License-Expression: MIT

{dragon_ml_toolbox-20.1.1 → dragon_ml_toolbox-20.2.0/dragon_ml_toolbox.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dragon-ml-toolbox
-Version: 20.1.1
+Version: 20.2.0
 Summary: Complete pipelines and helper tools for data science and machine learning projects.
 Author-email: Karl Luigi Loza Vidaurre <luigiloza@gmail.com>
 License-Expression: MIT

{dragon_ml_toolbox-20.1.1 → dragon_ml_toolbox-20.2.0}/ml_tools/ML_datasetmaster/_base_datasetmaster.py

@@ -172,6 +172,12 @@ class _BaseDatasetMaker(ABC):
         y_train_arr = y_train.to_numpy() if isinstance(y_train, (pandas.Series, pandas.DataFrame)) else y_train
         y_val_arr = y_val.to_numpy() if isinstance(y_val, (pandas.Series, pandas.DataFrame)) else y_val
         y_test_arr = y_test.to_numpy() if isinstance(y_test, (pandas.Series, pandas.DataFrame)) else y_test
+        
+        # --- Ensure targets are 2D (N, 1) if they are currently 1D (N,) ---
+        if y_train_arr.ndim == 1: y_train_arr = y_train_arr.reshape(-1, 1)
+        if y_val_arr.ndim == 1:   y_val_arr = y_val_arr.reshape(-1, 1)
+        if y_test_arr.ndim == 1:  y_test_arr = y_test_arr.reshape(-1, 1)
+        # ------------------------------------------------------------------
 
         if self.target_scaler is None:
             _LOGGER.info("Fitting a new DragonScaler on training targets.")

{dragon_ml_toolbox-20.1.1 → dragon_ml_toolbox-20.2.0}/ml_tools/ML_scaler/_ML_scaler.py

@@ -127,9 +127,14 @@ class DragonScaler:
         if self.mean_ is None or self.std_ is None or self.continuous_feature_indices is None:
             # If not fitted, return as is
             return data
-
+        
         data_clone = data.clone()
         
+        # Robust check: If data is 1D, assume it's a single feature/target column and reshape it to (N, 1) for the operation, then reshape back.
+        input_is_1d = (data_clone.ndim == 1)
+        if input_is_1d:
+            data_clone = data_clone.view(-1, 1)
+
         # Ensure mean and std are on the same device as the data
         mean = self.mean_.to(data.device)
         std = self.std_.to(data.device)
@@ -143,6 +148,9 @@ class DragonScaler:
         # Place the scaled features back into the cloned tensor
         data_clone[:, self.continuous_feature_indices] = scaled_features
         
+        if input_is_1d:
+            return data_clone.view(-1)
+        
         return data_clone
 
     def inverse_transform(self, data: torch.Tensor) -> torch.Tensor:
@@ -151,9 +159,13 @@ class DragonScaler:
         """
         if self.mean_ is None or self.std_ is None or self.continuous_feature_indices is None:
             return data
-            
+        
         data_clone = data.clone()
         
+        input_is_1d = (data_clone.ndim == 1)
+        if input_is_1d:
+            data_clone = data_clone.view(-1, 1)
+        
         mean = self.mean_.to(data.device)
         std = self.std_.to(data.device)
         
@@ -164,6 +176,9 @@ class DragonScaler:
         
         data_clone[:, self.continuous_feature_indices] = original_scale_features
         
+        if input_is_1d:
+            return data_clone.view(-1)
+        
         return data_clone
 
     def _get_state(self):

{dragon_ml_toolbox-20.1.1 → dragon_ml_toolbox-20.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "dragon-ml-toolbox"
-version = "20.1.1"
+version = "20.2.0"
 description = "Complete pipelines and helper tools for data science and machine learning projects."
 authors = [
     { name = "Karl Luigi Loza Vidaurre", email = "luigiloza@gmail.com" }