metacountregressor 0.1.126__tar.gz → 0.1.128__tar.gz

{metacountregressor-0.1.126 → metacountregressor-0.1.128}/PKG-INFO

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

{metacountregressor-0.1.126 → metacountregressor-0.1.128}/metacountregressor/helperprocess.py

@@ -183,18 +183,18 @@ config = {
 # Function to guess Low, Medium, High ranges
 def guess_low_medium_high(column_name, series):
     # Compute the tertiles (33rd and 66th percentiles)
-    low_threshold = series.quantile(0.33)
-    high_threshold = series.quantile(0.66)
+    low_threshold = np.quantile(series, 0.33)
+    high_threshold = np.quantile(series,0.66)
 
     # Define the bins and labels
-    bins = [series.min() - 1, low_threshold, high_threshold, series.max()]
+    bins = [np.min(series) - 1, low_threshold, high_threshold, np.min(series)]
     labels = ['Low', 'Medium', 'High']
 
     return {
         'type': 'bin',
         'bins': bins,
         'labels': labels,
-        'prefix': f'{column_name}_Binned'
+        'prefix': f'{column_name}'
     }
 
 def transform_dataframe(df, config):
@@ -246,8 +246,7 @@ def guess_column_type(column_name, series):
         if unique_values < 5:
             return {'type': 'one-hot', 'prefix': column_name}
 
-
-        elif series.max() - series.min() > 20:
+        elif np.max(series) - np.min(series) > 20:
             # If there are few unique values, assume binning with default bins
             guess_low_medium_high(column_name,series)
         else:

{metacountregressor-0.1.126 → metacountregressor-0.1.128}/metacountregressor/solution.py

@@ -449,7 +449,7 @@ class ObjectiveFunction(object):
         if 'model_types' in kwargs:
             model_types = kwargs['model_types']
         else:
-            print('the type of models possible are:')
+
 
             model_types = [[0, 1]]  # add 2 for Generalized Poisson
             model_types = [[0]]
@@ -459,7 +459,7 @@ class ObjectiveFunction(object):
         # Retrieve the keys (model names) corresponding to the values in model_types
         model_keys = [key for key, value in model_t_dict.items() if value in model_types[0]]
         # Print the formatted result
-        print(f'The type of models possible are: {", ".join(model_keys)}')
+        print(f'The type of models possible will consider: {", ".join(model_keys)}')
         self._discrete_values = self._discrete_values + self.define_poissible_transforms(
             self._transformations, kwargs.get('decisions',None)) + model_types
 

{metacountregressor-0.1.126 → metacountregressor-0.1.128}/metacountregressor.egg-info/PKG-INFO

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