metacountregressor 0.1.149__py3-none-any.whl → 0.1.151__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -302,16 +302,25 @@ def transform_dataframe(df, config):
302
302
  for column, settings in config.items():
303
303
  if settings['type'] == 'bin':
304
304
  # Apply binning
305
- binned = pd.cut(
306
- binned = df[column],
307
- bins=settings['bins'],
308
- labels=settings['labels'],
309
- right=False,
310
- duplicates='drop'
305
+ # Get unique bins (remove duplicates)
306
+ unique_bins = sorted(set(settings['bins']))
311
307
 
308
+ # Adjust labels if necessary
309
+ if len(unique_bins) - 1 != len(settings['labels']):
310
+ print(f"Adjusting labels to match bins: {len(unique_bins) - 1} bins detected.")
311
+ labels = [f'Bin {i+1}' for i in range(len(unique_bins) - 1)]
312
+ else:
313
+ labels = settings['labels']
314
+
315
+ # Perform the binning
316
+ binned_d = pd.cut(
317
+ df[column],
318
+ bins=unique_bins, # Deduplicated bins
319
+ labels=labels, # Adjusted or original labels
320
+ right=False # Adjust based on whether to include the right edge
312
321
  )
313
322
  # One-hot encode the binned column
314
- binned_dummies = pd.get_dummies(binned, prefix=settings['prefix'])
323
+ binned_dummies = pd.get_dummies(binned_d, prefix=settings['prefix'])
315
324
  output_df = pd.concat([output_df, binned_dummies], axis=1)
316
325
 
317
326
  elif settings['type'] == 'one-hot':
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: metacountregressor
3
- Version: 0.1.149
3
+ Version: 0.1.151
4
4
  Summary: Extensions for a Python package for estimation of count models.
5
5
  Home-page: https://github.com/zahern/CountDataEstimation
6
6
  Author: Zeke Ahern
@@ -3,7 +3,7 @@ metacountregressor/_device_cust.py,sha256=759fnKmTYccJm4Lpi9_1reurh6OB9d6q9soPR0
3
3
  metacountregressor/app_main.py,sha256=vY3GczTbGbBRalbzMkl_9jVW7RMgEOc6z2Dr1IZJv9c,10014
4
4
  metacountregressor/data_split_helper.py,sha256=M2fIMdIO8znUaYhx5wlacRyNWdQjNYu1z1wkE-kFUYU,3373
5
5
  metacountregressor/halton.py,sha256=jhovA45UBoZYU9g-hl6Lb2sBIx_ZBTNdPrpgkzR9fng,9463
6
- metacountregressor/helperprocess.py,sha256=2Wy_JirQIBQLIxweGegF8mbudY9qAm-M-THZKrHZrpc,21115
6
+ metacountregressor/helperprocess.py,sha256=0cA9_kRRyDGR_aGuClzHyHPsn3iwriH6PD8l7T2jCAo,21653
7
7
  metacountregressor/main.py,sha256=2Rx_mGIGzl4lhwkMb7DHvsBaawqEakKiVR1Yr2uG9Yo,22819
8
8
  metacountregressor/main_old.py,sha256=eTS4ygq27MnU-dZ_j983Ucb-D5XfbVF8OJQK2hVVLZc,24123
9
9
  metacountregressor/metaheuristics.py,sha256=1Tm_BS8HxumEo9fteIwTTkfiJXbfH6wlfSmY0tSt77A,106289
@@ -13,8 +13,8 @@ metacountregressor/setup.py,sha256=5UcQCCLR8Fm5odA3MX78WwahavxFq4mVD6oq0IuQvAY,9
13
13
  metacountregressor/single_objective_finder.py,sha256=jVG7GJBqzSP4_riYr-kMMKy_LE3SlGmKMunNhHYxgRg,8011
14
14
  metacountregressor/solution.py,sha256=Se3-NMxuRFacPN9f3qfGzkKeIccs8cbe4ebkV0UsGlA,278325
15
15
  metacountregressor/test_generated_paper2.py,sha256=pwOoRzl1jJIIOUAAvbkT6HmmTQ81mwpsshn9SLdKOg8,3927
16
- metacountregressor-0.1.149.dist-info/LICENSE.txt,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
17
- metacountregressor-0.1.149.dist-info/METADATA,sha256=fc7oSeeMlce7jL3-oTHM2n5hmRiHa_yA3gzlZS1XAMo,23434
18
- metacountregressor-0.1.149.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
19
- metacountregressor-0.1.149.dist-info/top_level.txt,sha256=zGG7UC5WIpr76gsFUpwJ4En2aCcoNTONBaS3OewwjR0,19
20
- metacountregressor-0.1.149.dist-info/RECORD,,
16
+ metacountregressor-0.1.151.dist-info/LICENSE.txt,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
17
+ metacountregressor-0.1.151.dist-info/METADATA,sha256=UnAwgc0WSRSryWmpvJhJrIOC0YkUfq6xrgtdqRBPw_M,23434
18
+ metacountregressor-0.1.151.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
19
+ metacountregressor-0.1.151.dist-info/top_level.txt,sha256=zGG7UC5WIpr76gsFUpwJ4En2aCcoNTONBaS3OewwjR0,19
20
+ metacountregressor-0.1.151.dist-info/RECORD,,