metacountregressor 0.1.138__py3-none-any.whl → 0.1.139__py3-none-any.whl

metacountregressor/helperprocess.py

@@ -1,9 +1,12 @@
+from os.path import exists
+
 import numpy as np
 import pandas as pd
 import csv
 import matplotlib.pyplot as plt
 from scipy import stats as st
 from sklearn.preprocessing import StandardScaler
+from win32comext.shell.demos.IActiveDesktop import existing_item
 
 plt.style.use('https://github.com/dhaitz/matplotlib-stylesheets/raw/master/pitayasmoothie-dark.mplstyle')
 
@@ -179,17 +182,38 @@ config = {
     }
 }
 '''
-def set_up_analyst_constraints(data_characteristic, variable_decisions_alt = None):
+def null_handler(vari):
+    if vari in locals():
+        return vari
+    else:
+        print(f'{vari} does not exist, setting None..')
+        return None
+
+
+def set_up_analyst_constraints(data_characteristic, model_terms,  variable_decisions_alt = None):
+
+
     name_data_characteristics = data_characteristic.columns.tolist()
-    distu = ['n', 'u', 't']
-    tra = ['no']
+    # Get non-None values as a list
+    non_none_terms = [value for value in model_terms.values() if value is not None]
+    # how to make name_data_characteristics - non_none_terms
+
+    result = [item for item in name_data_characteristics if item not in non_none_terms]
+    distu = ['Normal', 'Uniform', 'Triangular']
+    tra = ['no', 'sqrt', 'arcsinh']
+    if model_terms.get('grouped') is None:
+        print('cant have grouped rpm, removing level 4 from every item')
+        MAKE_ALL_4_FALSE = True
+    else:
+        MAKE_ALL_4_FALSE = False
+
     variable_decisions = {
         name: {
             'levels': list(range(6)),
             'distributions': distu,
             'transformations': tra
         }
-        for name in name_data_characteristics
+        for name in result
     }
     # Override elements in the original dictionary with the alt dictionary
     if variable_decisions_alt is not None:
@@ -208,7 +232,11 @@ def set_up_analyst_constraints(data_characteristic, variable_decisions_alt = Non
 
         # Add levels as True/False for Level 0 through Level 5
         for level in range(6):  # Assuming Level 0 to Level 5
-            row[f'Level {level}'] = level in details['levels']
+
+            if level == 4 and MAKE_ALL_4_FALSE:
+                row[f'Level {level}'] = False
+            else:
+                row[f'Level {level}'] = level in details['levels']
 
         # Add distributions and transformations directly
         row['distributions'] = details['distributions']
@@ -218,7 +246,9 @@ def set_up_analyst_constraints(data_characteristic, variable_decisions_alt = Non
 
     # Create the DataFrame
     df = pd.DataFrame(rows)
-    return  df
+
+    data_new = data_characteristic.rename(columns={v: k for k, v in model_terms.items() if v in data_characteristic.columns})
+    return  df, data_new
 
 # Function to guess Low, Medium, High ranges
 def guess_low_medium_high(column_name, series):

{metacountregressor-0.1.138.dist-info → metacountregressor-0.1.139.dist-info}/METADATA

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

{metacountregressor-0.1.138.dist-info → metacountregressor-0.1.139.dist-info}/RECORD

@@ -3,7 +3,7 @@ metacountregressor/_device_cust.py,sha256=759fnKmTYccJm4Lpi9_1reurh6OB9d6q9soPR0
 metacountregressor/app_main.py,sha256=vY3GczTbGbBRalbzMkl_9jVW7RMgEOc6z2Dr1IZJv9c,10014
 metacountregressor/data_split_helper.py,sha256=M2fIMdIO8znUaYhx5wlacRyNWdQjNYu1z1wkE-kFUYU,3373
 metacountregressor/halton.py,sha256=jhovA45UBoZYU9g-hl6Lb2sBIx_ZBTNdPrpgkzR9fng,9463
-metacountregressor/helperprocess.py,sha256=3SwQtKCMwu2_6pna1L_vM_qUJiuqi07HnWgHAned008,19502
+metacountregressor/helperprocess.py,sha256=WiCItpiGJQJNxwczsNzWobDeltfmSKzSJKlxz8_9pxk,20504
 metacountregressor/main.py,sha256=2Rx_mGIGzl4lhwkMb7DHvsBaawqEakKiVR1Yr2uG9Yo,22819
 metacountregressor/main_old.py,sha256=eTS4ygq27MnU-dZ_j983Ucb-D5XfbVF8OJQK2hVVLZc,24123
 metacountregressor/metaheuristics.py,sha256=Kkx1Jfox6NBlm5zVrI26Vc_NI7NFQSS9dinrZU9SpV8,105871
@@ -13,8 +13,8 @@ metacountregressor/setup.py,sha256=5UcQCCLR8Fm5odA3MX78WwahavxFq4mVD6oq0IuQvAY,9
 metacountregressor/single_objective_finder.py,sha256=jVG7GJBqzSP4_riYr-kMMKy_LE3SlGmKMunNhHYxgRg,8011
 metacountregressor/solution.py,sha256=6qAtCSKNvmrCpIpBCzp2Zt8wWKZBHRQtTriyiMHUo54,277519
 metacountregressor/test_generated_paper2.py,sha256=pwOoRzl1jJIIOUAAvbkT6HmmTQ81mwpsshn9SLdKOg8,3927
-metacountregressor-0.1.138.dist-info/LICENSE.txt,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-metacountregressor-0.1.138.dist-info/METADATA,sha256=IJASEKH3eOWCUVWj1W1_yi5MoLbEaU6AN6_9cm5NHwo,23434
-metacountregressor-0.1.138.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-metacountregressor-0.1.138.dist-info/top_level.txt,sha256=zGG7UC5WIpr76gsFUpwJ4En2aCcoNTONBaS3OewwjR0,19
-metacountregressor-0.1.138.dist-info/RECORD,,
+metacountregressor-0.1.139.dist-info/LICENSE.txt,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+metacountregressor-0.1.139.dist-info/METADATA,sha256=_jvPMVa-G1tSUPlH9VjpCAPT1nl6TCFWSsq6SMDwTw8,23434
+metacountregressor-0.1.139.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+metacountregressor-0.1.139.dist-info/top_level.txt,sha256=zGG7UC5WIpr76gsFUpwJ4En2aCcoNTONBaS3OewwjR0,19
+metacountregressor-0.1.139.dist-info/RECORD,,