lifejacket 1.0.0__py3-none-any.whl → 1.0.2__py3-none-any.whl

lifejacket/small_sample_corrections.py

@@ -18,20 +18,18 @@ def perform_desired_small_sample_correction(
     small_sample_correction,
     per_subject_joint_adjusted_meat_contributions,
     per_subject_classical_meat_contributions,
-    per_subject_classical_bread_inverse_contributions,
+    per_subject_classical_bread_contributions,
     num_subjects,
     theta_dim,
 ):
 
-    # We first compute the classical inverse bread matrix and invert it.  While
+    # We first compute the classical bread matrix and invert it.  While
     # it is possible to avoid this inversion using a QR decomposition and
     # solving linear systems (discussed more below), we typically don't have
     # issues with the conditioning of just the classical bread.
-    classical_bread_inverse_matrix = jnp.mean(
-        per_subject_classical_bread_inverse_contributions, axis=0
-    )
+    classical_bread_matrix = jnp.mean(per_subject_classical_bread_contributions, axis=0)
     classical_bread_matrix = invert_matrix_and_check_conditioning(
-        classical_bread_inverse_matrix,
+        classical_bread_matrix,
     )[0]
 
     # These will hold either corrective matrices or scalar weights depending on
@@ -43,9 +41,9 @@ def perform_desired_small_sample_correction(
     per_subject_classical_correction_weights = np.ones(num_subjects)
     if small_sample_correction == SmallSampleCorrections.NONE:
         logger.info(
-            "No small sample correction requested. Using the raw per-subject joint adjusted bread inverse contributions."
+            "No small sample correction requested. Using the raw per-subject joint adjusted bread contributions."
         )
-    elif small_sample_correction == SmallSampleCorrections.HC1theta:
+    elif small_sample_correction == SmallSampleCorrections.Z1theta:
         logger.info(
             "Using HC1 small sample correction at the subject trajectory level. Note that we are treating the number of parameters as simply the size of theta, despite the presence of betas."
         )
@@ -53,12 +51,12 @@ def perform_desired_small_sample_correction(
             per_subject_classical_correction_weights
         ) = (num_subjects / (num_subjects - theta_dim) * np.ones(num_subjects))
     elif small_sample_correction in {
-        SmallSampleCorrections.HC2theta,
-        SmallSampleCorrections.HC3theta,
+        SmallSampleCorrections.Z2theta,
+        SmallSampleCorrections.Z3theta,
     }:
         logger.info("Using %s small sample correction at the subject trajectory level.")
 
-        power = 1 if small_sample_correction == SmallSampleCorrections.HC2theta else 2
+        power = 1 if small_sample_correction == SmallSampleCorrections.Z2theta else 2
 
         # It turns out to typically not make sense to compute the adjusted analog
         # of the classical leverages, since this involves correcting the joint adjusted meat matrix
@@ -75,12 +73,12 @@ def perform_desired_small_sample_correction(
 
         # TODO: Write a unit test for some level of logic here and then rewrite this to not require
         # the classical bread explicitly. May be slower, probably needs a for loop so that can use
-        # a solver for each matrix multiplication after a QR decomposition of the bread inverse
+        # a solver for each matrix multiplication after a QR decomposition of the bread
         # transpose.
         classical_leverages_per_subject = (
             np.einsum(
                 "nij,ji->n",
-                per_subject_classical_bread_inverse_contributions,
+                per_subject_classical_bread_contributions,
                 classical_bread_matrix,
             )
             / num_subjects

{lifejacket-1.0.0.dist-info → lifejacket-1.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lifejacket
-Version: 1.0.0
+Version: 1.0.2
 Summary: Consistent standard errors for longitudinal data collected under pooling online decision policies.
 Author-email: Nowell Closser <nowellclosser@gmail.com>
 Requires-Python: >=3.10

lifejacket-1.0.2.dist-info/RECORD

@@ -0,0 +1,17 @@
+lifejacket/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lifejacket/arg_threading_helpers.py,sha256=jiPdG-1USYUXqXAEfRPL-3KBKno7uMr4-fCl9GrZ7Fw,17599
+lifejacket/calculate_derivatives.py,sha256=Jv-N54lbUSi36H9hTKtupy0ReoXKzI8mFlgqmQddL5U,37476
+lifejacket/constants.py,sha256=cQjA2-YRwsGDMjsndkbFK7m56xbVpUo-XCyyg9gY1BA,313
+lifejacket/deployment_conditioning_monitor.py,sha256=Go4YhJZfkwj9g0mP0vYCVROKS397CIviHUtprdLNYTk,43813
+lifejacket/form_adjusted_meat_adjustments_directly.py,sha256=AVlGOuw_FgVDcVnhQs1GorxtBMKUXvmYCQZgqG675k4,13539
+lifejacket/get_datum_for_blowup_supervised_learning.py,sha256=sCH-PlrFlLJgCYpTmdeasiHwHYSEy9wxspkOTDuDPuY,58594
+lifejacket/helper_functions.py,sha256=SdAbUwXNx-3JFsyTfLyliQ7kUOm0eABaiNgoYLR8NG0,16967
+lifejacket/input_checks.py,sha256=q7HFZq5n18edQU8X5laONsBgWSMidLRy6Nhqdw5FpOw,47084
+lifejacket/post_deployment_analysis.py,sha256=XdKObve0hOXVwPWSDD2lEEfrWAdcZK-c-uh53HIrKLM,82664
+lifejacket/small_sample_corrections.py,sha256=aB6qi-r3ANoBMgf2Oo5-lCXCy_L4H3FlBffGwPcfXkg,5610
+lifejacket/vmap_helpers.py,sha256=pZqYN3p9Ty9DPOeeY9TKbRJXR2AV__HBwwDFOvdOQ84,2688
+lifejacket-1.0.2.dist-info/METADATA,sha256=L48u8IMsEXMTwtBJlFwh4mJ-pZUD8NcQEuqgjV0zkjo,1773
+lifejacket-1.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lifejacket-1.0.2.dist-info/entry_points.txt,sha256=CZ9AUPN0xfnpYqwtGTr6n9l5mpyEOddsXX8fnxKRB6U,71
+lifejacket-1.0.2.dist-info/top_level.txt,sha256=vKl8m7jOQ4pkbzVuHCJsq-8LcXRrOAWnok3bBo9qpsE,11
+lifejacket-1.0.2.dist-info/RECORD,,

lifejacket-1.0.2.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+lifejacket = lifejacket.post_deployment_analysis:cli

lifejacket-1.0.0.dist-info/RECORD

@@ -1,17 +0,0 @@
-lifejacket/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lifejacket/after_study_analysis.py,sha256=-6LCfl0dz9eHkP5aLs-J2YqpxVJ57J8Ov4wEWKbZMWA,83315
-lifejacket/arg_threading_helpers.py,sha256=jiPdG-1USYUXqXAEfRPL-3KBKno7uMr4-fCl9GrZ7Fw,17599
-lifejacket/calculate_derivatives.py,sha256=wFMS1pSm5MI8D7FYK2nPEAdvtVJAxKy8OVtUj7giII0,37520
-lifejacket/constants.py,sha256=2L05p6NJ7l3qRZ1hD2KlrvzWF1ReSmWRUkULPIhdvJo,842
-lifejacket/deployment_conditioning_monitor.py,sha256=5PMBfDQfyxwoYXr3qh1pTdMBefnSjjQMFyCj3_sItVY,43526
-lifejacket/form_adjusted_meat_adjustments_directly.py,sha256=ML7FOB_hnDEM1ndC2X4j_oaqdqn-EPr9j-MetlOqSE8,13740
-lifejacket/get_datum_for_blowup_supervised_learning.py,sha256=SjA3H7H8RBZc9GxzgsccqZoLZnAD68mxaPH1E87s2lA,59092
-lifejacket/helper_functions.py,sha256=ldkanFn5b2lwBlni-HtXPVd7QvxEKmzL1UsJMsG6f6g,23478
-lifejacket/input_checks.py,sha256=RhS2tnq74jOU5_pVSXsxthSmuGhlmoMc5eJiourQiS0,47513
-lifejacket/small_sample_corrections.py,sha256=bnbtxjEKZBiFrCSF2WA0vczaVeIqao1NekxJRZdmwCU,5692
-lifejacket/vmap_helpers.py,sha256=pZqYN3p9Ty9DPOeeY9TKbRJXR2AV__HBwwDFOvdOQ84,2688
-lifejacket-1.0.0.dist-info/METADATA,sha256=UGZsalPpzX8K7BnstZ0RDFq-1ZpL5CsXcUd_hYvQWus,1773
-lifejacket-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lifejacket-1.0.0.dist-info/entry_points.txt,sha256=4k8ibVIUT-OHxPaaDv-QwWpC64ErzhdemHpTAXCnb8w,67
-lifejacket-1.0.0.dist-info/top_level.txt,sha256=vKl8m7jOQ4pkbzVuHCJsq-8LcXRrOAWnok3bBo9qpsE,11
-lifejacket-1.0.0.dist-info/RECORD,,

lifejacket-1.0.0.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-lifejacket = lifejacket.after_study_analysis:cli