metacountregressor 0.1.148__tar.gz → 0.1.149__tar.gz

{metacountregressor-0.1.148 → metacountregressor-0.1.149}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: metacountregressor
-Version: 0.1.148
+Version: 0.1.149
 Summary: Extensions for a Python package for estimation of count models.
 Home-page: https://github.com/zahern/CountDataEstimation
 Author: Zeke Ahern

{metacountregressor-0.1.148 → metacountregressor-0.1.149}/metacountregressor/helperprocess.py

@@ -302,11 +302,22 @@ def transform_dataframe(df, config):
     for column, settings in config.items():
         if settings['type'] == 'bin':
             # Apply binning
+            # Get unique bins (remove duplicates)
+            unique_bins = sorted(set(settings['bins']))
+
+            # Adjust labels if necessary
+            if len(unique_bins) - 1 != len(settings['labels']):
+                print(f"Adjusting labels to match bins: {len(unique_bins) - 1} bins detected.")
+                labels = [f'Bin {i+1}' for i in range(len(unique_bins) - 1)]
+            else:
+                labels = settings['labels']
+
+            # Perform the binning
             binned_d = pd.cut(
                 df[column],
-                bins=sorted(set(settings['bins'])),  # Remove duplicate bins
-                labels=settings['labels'],
-                right=False  # Adjust based on your requirements
+                bins=unique_bins,  # Deduplicated bins
+                labels=labels,     # Adjusted or original labels
+                right=False        # Adjust based on whether to include the right edge
             )
             # One-hot encode the binned column
             binned_dummies = pd.get_dummies(binned_d, prefix=settings['prefix'])

{metacountregressor-0.1.148 → metacountregressor-0.1.149}/metacountregressor.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: metacountregressor
-Version: 0.1.148
+Version: 0.1.149
 Summary: Extensions for a Python package for estimation of count models.
 Home-page: https://github.com/zahern/CountDataEstimation
 Author: Zeke Ahern