metacountregressor 0.1.147__tar.gz → 0.1.149__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: metacountregressor
-Version: 0.1.147
+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.147 → metacountregressor-0.1.149}/metacountregressor/helperprocess.py

@@ -302,16 +302,25 @@ def transform_dataframe(df, config):
     for column, settings in config.items():
         if settings['type'] == 'bin':
             # Apply binning
-            binned = pd.cut(
-                binned = df[column],
-                bins=settings['bins'],
-                labels=settings['labels'],
-                right=False,
-                duplicates='drop'
+            # 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=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, prefix=settings['prefix'])
+            binned_dummies = pd.get_dummies(binned_d, prefix=settings['prefix'])
             output_df = pd.concat([output_df, binned_dummies], axis=1)
 
         elif settings['type'] == 'one-hot':

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: metacountregressor
-Version: 0.1.147
+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