metacountregressor 0.1.315__py3-none-any.whl → 0.1.317__py3-none-any.whl

metacountregressor/solution.py

@@ -459,8 +459,14 @@ class ObjectiveFunction(object):
 
         if self.G is not None:
             #TODO need to handle this for groups
-            self._distribution = ["trad| " + item for item in self._distribution
+            print('group by groups off for now')
+            NAH_GROUP = 0
+            if NAH_GROUP:
+                self._distribution = ["trad| " + item for item in self._distribution
                                   ] + ["grpd| " + item for item in self._distribution]
+            else:
+                self._distribution = ["trad| " + item for item in self._distribution
+                                  ]
 
         # output information
         self.convergence = None
@@ -1035,9 +1041,9 @@ class ObjectiveFunction(object):
 
                 abct = []
                 hetro_long = hetro_long + hetro_std
-                for i in model_nature['transfrom_hetro']:
+                for i in model_nature['transform_hetro']:
                     abct = abct + i
-                for i in model_nature['transfrom_hetro']:
+                for i in model_nature['transform_hetro']:
                     abct = abct + ['']
             else:
                 big_hetro = []
@@ -3563,9 +3569,9 @@ class ObjectiveFunction(object):
     def _chol_mat(self, correlationLength, br, Br_w, correlation):
         # if correlation = True correlation pos is randpos, if list get correct pos
         
+        print(self.rdm_grouped_fit, 'i think this causes the errror')
         
-        
-        varnames = self.none_handler(
+        varnames = self.none_handler(self.grouped_rpm)+ self.none_handler(
                 self.rdm_fit) + self.none_handler(self.rdm_cor_fit)
 
         
@@ -4874,11 +4880,17 @@ class ObjectiveFunction(object):
                 n_coeff = self.get_param_num(dispersion)
                 Kf_a, Kr_a, Kr_c, Kr_b_a, Kchol_a, Kh = self.get_num_params()
                 if Kchol_a != Kchol:
-                    print('this should not hh qhy')
+                    print('this should not happeb but why', Kr_b, 'kr)b is')
 
                 if Kr_b != Kr_a:
-                    print('hold qhy this should never happen')
+                    print('self.rdm_cor_fit', self.rdm_cor_fit)
+                    print('grouped_fit', self.rdm_grouped_fit)
+                    print('self.rdm_fit', self.rdm_fit) 
+                    print('odd, check this this should never happen')
+                
+
 
+                self.sanity_check(Xr, self.grouped_rpm)
 
 
 
@@ -4971,7 +4983,8 @@ class ObjectiveFunction(object):
                     betas_hetro = betas[Kf + Kr + Kr_b + Kchol:Kf + Kr + Kr_b + Kchol + KFH]
                     # betas_hetro = betas[:Xdh.shape[2]+1]
                     betas_hetro_sd = betas[Kf + Kr + Kr_b + Kchol + KFH:Kf + Kr + Kr_b + Kchol + KFH + KFHs]
-                    if KFHs > 1:
+
+                    if KFHs >= 1:
                         # print('now what how do i split')
                         x_i_h = model_nature.get('x_h_storage_test') if test_set else model_nature.get('x_h_storage')
                         if test_set == 2:
@@ -4992,6 +5005,7 @@ class ObjectiveFunction(object):
                             Vdh += dev.cust_einsum("njk,nkr -> njr", i, Bh)
 
                     else:
+                        print('this will not work cause i cant trul the right drawse')
                         Bh = self._transform_rand_betas(betas_hetro, betas_hetro_sd, draws_hetro)
 
                         Vdh = dev.cust_einsum("njk,nkr -> njr", Xdh, Bh)
@@ -5971,13 +5985,13 @@ class ObjectiveFunction(object):
             Solution object with updated objectives.
         """
         # Extract relevant data
-        X, Xr, XG = mod.get('X'), mod.get('Xr'), mod.get('XG')
+        X, Xr, XG, XH= mod.get('X'), mod.get('Xr'), mod.get('XG'), mod.get('XH')
         distribution = mod.get('dist_fit')
 
         # Prepare draws
         draws = self._prepare_draws(Xr, distribution)
         draws_grouped = self._prepare_grouped_draws(XG, mod) if XG is not None else None
-
+        mod = self._prepare_hetro(mod)
         # Optimization method and options
         method = self.method_ll if bounds is None else 'L-BFGS-B'
         
@@ -6451,6 +6465,27 @@ class ObjectiveFunction(object):
         return self.prepare_halton(
             n_random_effects, n_samples, self.Ndraws, distribution, long=False, slice_this_way=self.group_halton
         )
+    
+    def _prepare_hetro(self, mod):
+        if 'XH' in mod and len(mod.get('hetro_hold')) > 0:
+
+            XH = mod.get('XH')
+
+            styd = list(mod.get('hetro_hold').keys())
+
+            nh, ph, ______ = XH.shape
+            kgh = len(mod.get('hetro_hold'))
+            draws_hetro = self.prepare_halton(kgh, nh, self.Ndraws, styd, slice_this_way=self.group_halton)
+            mod['draws_hetro'] = draws_hetro.copy()
+            if self.is_multi:
+                XHtest = mod.get('XH_test')
+                nht, pht, ______ = XHtest.shape
+                draws_hetro_test = self.prepare_halton(kgh, nht, self.Ndraws, styd,
+                                                    slice_this_way=self.group_halton_test)
+                mod['draws_hetro_test'] = draws_hetro_test.copy()
+        return mod
+
+
 
     def _prepare_grouped_draws(self, XG, mod):
         """
@@ -7216,8 +7251,12 @@ class ObjectiveFunction(object):
         indices = [i for i, name in enumerate(self._characteristics_names) if name in names and isinstance(name, str)]
         return indices
 
-    def sanity_check(self, Xr):
-        a =len(self.rdm_cor_fit)+len(self.rdm_fit) != Xr.shape[2]
+    def sanity_check(self, Xr, grouped_rpm = None):
+        if grouped_rpm is None:
+            a =len(self.rdm_cor_fit)+len(self.rdm_fit) != Xr.shape[2]
+        else:
+            a = len(self.none_handler(self.rdm_cor_fit)) + len(self.none_handler(self.rdm_fit)) + len(self.none_handler(grouped_rpm)) != Xr.shape[2]
+            
         if a:
             print('why')
             print('The number of random effects does not match the data')
@@ -7291,7 +7330,7 @@ class ObjectiveFunction(object):
                 X_h_test = df_test[:, :, indices_hetro]
                 x_h_storage_test.append(X_h_test)
         model_nature['x_h_storage'] = x_h_storage
-        model_nature['transfrom_hetro'] = transform_hetro
+        model_nature['transform_hetro'] = transform_hetro
         model_nature['x_h_storage_test'] = x_h_storage_test
 
         if hasattr(self, 'group_dummies'):

{metacountregressor-0.1.315.dist-info → metacountregressor-0.1.317.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: metacountregressor
-Version: 0.1.315
+Version: 0.1.317
 Summary: Extensive Testing for Estimation of Data Count Models
 Home-page: https://github.com/zahern/CountDataEstimation
 Author: Zeke Ahern

{metacountregressor-0.1.315.dist-info → metacountregressor-0.1.317.dist-info}/RECORD

@@ -11,10 +11,10 @@ metacountregressor/pareto_file.py,sha256=whySaoPAUWYjyI8zo0hwAOa3rFk6SIUlHSpqZiL
 metacountregressor/pareto_logger__plot.py,sha256=mEU2QN4wmsM7t39GJ_XhJ_jjsdl09JOmG0U2jICrAkI,30037
 metacountregressor/setup.py,sha256=5UcQCCLR8Fm5odA3MX78WwahavxFq4mVD6oq0IuQvAY,936
 metacountregressor/single_objective_finder.py,sha256=jVG7GJBqzSP4_riYr-kMMKy_LE3SlGmKMunNhHYxgRg,8011
-metacountregressor/solution.py,sha256=r9BSb6Uh0JPenn675MYHexXvol6JQy_IguAN9WnQCCo,320049
+metacountregressor/solution.py,sha256=ap9HscK_9N1m3jpgJju75Y7LLxZ8Kxa-Pkcg1hBnkZI,321872
 metacountregressor/test_generated_paper2.py,sha256=pwOoRzl1jJIIOUAAvbkT6HmmTQ81mwpsshn9SLdKOg8,3927
-metacountregressor-0.1.315.dist-info/licenses/LICENSE.txt,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-metacountregressor-0.1.315.dist-info/METADATA,sha256=6CFlRUuRRFPIEld3n5paplu2wszAgYGl3_Mn-FAXLZ0,23581
-metacountregressor-0.1.315.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-metacountregressor-0.1.315.dist-info/top_level.txt,sha256=zGG7UC5WIpr76gsFUpwJ4En2aCcoNTONBaS3OewwjR0,19
-metacountregressor-0.1.315.dist-info/RECORD,,
+metacountregressor-0.1.317.dist-info/licenses/LICENSE.txt,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+metacountregressor-0.1.317.dist-info/METADATA,sha256=eaozI5zX4AjG-rz0NJ6b8BvpI46FehiiVzq8yYbtTGk,23581
+metacountregressor-0.1.317.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+metacountregressor-0.1.317.dist-info/top_level.txt,sha256=zGG7UC5WIpr76gsFUpwJ4En2aCcoNTONBaS3OewwjR0,19
+metacountregressor-0.1.317.dist-info/RECORD,,