@@ -100,6 +100,7 @@ <h1>Source code for dte_adj.local</h1><div class="highlight"><pre>
100100 < span class ="n "> control_treatment_arm</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span >
101101 < span class ="n "> locations</ span > < span class ="p "> :</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span >
102102 < span class ="n "> alpha</ span > < span class ="p "> :</ span > < span class ="nb "> float</ span > < span class ="o "> =</ span > < span class ="mf "> 0.05</ span > < span class ="p "> ,</ span >
103+ < span class ="n "> display_progress</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> True</ span > < span class ="p "> ,</ span >
103104 < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> Tuple</ span > < span class ="p "> [</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ]:</ span >
104105< span class ="w "> </ span > < span class ="sd "> """</ span >
105106< span class ="sd "> Compute Local Distribution Treatment Effects (LDTE).</ span >
@@ -113,6 +114,7 @@ <h1>Source code for dte_adj.local</h1><div class="highlight"><pre>
113114< span class ="sd "> control_treatment_arm (int): The index of the treatment arm of the control group.</ span >
114115< span class ="sd "> locations (np.ndarray): Scalar values to be used for computing the cumulative distribution.</ span >
115116< span class ="sd "> alpha (float, optional): Significance level of the confidence bound. Defaults to 0.05.</ span >
117+ < span class ="sd "> display_progress (bool, optional): Whether to display a progress bar. Defaults to True.</ span >
116118
117119< span class ="sd "> Returns:</ span >
118120< span class ="sd "> Tuple[np.ndarray, np.ndarray, np.ndarray]: A tuple containing:</ span >
@@ -156,6 +158,7 @@ <h1>Source code for dte_adj.local</h1><div class="highlight"><pre>
156158 < span class ="n "> control_treatment_arm</ span > < span class ="p "> ,</ span >
157159 < span class ="n "> locations</ span > < span class ="p "> ,</ span >
158160 < span class ="n "> alpha</ span > < span class ="p "> ,</ span >
161+ < span class ="n "> display_progress</ span > < span class ="p "> ,</ span >
159162 < span class ="p "> )</ span > </ div >
160163
161164
@@ -167,6 +170,7 @@ <h1>Source code for dte_adj.local</h1><div class="highlight"><pre>
167170 < span class ="n "> control_treatment_arm</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span >
168171 < span class ="n "> locations</ span > < span class ="p "> :</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span >
169172 < span class ="n "> alpha</ span > < span class ="p "> :</ span > < span class ="nb "> float</ span > < span class ="o "> =</ span > < span class ="mf "> 0.05</ span > < span class ="p "> ,</ span >
173+ < span class ="n "> display_progress</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> True</ span > < span class ="p "> ,</ span >
170174 < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> Tuple</ span > < span class ="p "> [</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ]:</ span >
171175< span class ="w "> </ span > < span class ="sd "> """</ span >
172176< span class ="sd "> Compute Local Probability Treatment Effects (LPTE).</ span >
@@ -181,6 +185,7 @@ <h1>Source code for dte_adj.local</h1><div class="highlight"><pre>
181185< span class ="sd "> locations (np.ndarray): Scalar values defining interval boundaries for probability computation.</ span >
182186< span class ="sd "> For each interval (locations[i], locations[i+1]], the LPTE is computed.</ span >
183187< span class ="sd "> alpha (float, optional): Significance level of the confidence bound. Defaults to 0.05.</ span >
188+ < span class ="sd "> display_progress (bool, optional): Whether to display a progress bar. Defaults to True.</ span >
184189
185190< span class ="sd "> Returns:</ span >
186191< span class ="sd "> Tuple[np.ndarray, np.ndarray, np.ndarray]: A tuple containing:</ span >
@@ -226,6 +231,7 @@ <h1>Source code for dte_adj.local</h1><div class="highlight"><pre>
226231 < span class ="n "> control_treatment_arm</ span > < span class ="p "> ,</ span >
227232 < span class ="n "> locations</ span > < span class ="p "> ,</ span >
228233 < span class ="n "> alpha</ span > < span class ="p "> ,</ span >
234+ < span class ="n "> display_progress</ span > < span class ="p "> ,</ span >
229235 < span class ="p "> )</ span > </ div >
230236</ div >
231237
@@ -281,6 +287,7 @@ <h1>Source code for dte_adj.local</h1><div class="highlight"><pre>
281287 < span class ="n "> control_treatment_arm</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span >
282288 < span class ="n "> locations</ span > < span class ="p "> :</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span >
283289 < span class ="n "> alpha</ span > < span class ="p "> :</ span > < span class ="nb "> float</ span > < span class ="o "> =</ span > < span class ="mf "> 0.05</ span > < span class ="p "> ,</ span >
290+ < span class ="n "> display_progress</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> True</ span > < span class ="p "> ,</ span >
284291 < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> Tuple</ span > < span class ="p "> [</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ]:</ span >
285292< span class ="w "> </ span > < span class ="sd "> """</ span >
286293< span class ="sd "> Compute Local Distribution Treatment Effects (LDTE) using ML adjustment.</ span >
@@ -293,6 +300,7 @@ <h1>Source code for dte_adj.local</h1><div class="highlight"><pre>
293300< span class ="sd "> control_treatment_arm (int): The index of the treatment arm of the control group.</ span >
294301< span class ="sd "> locations (np.ndarray): Scalar values to be used for computing the cumulative distribution.</ span >
295302< span class ="sd "> alpha (float, optional): Significance level of the confidence bound. Defaults to 0.05.</ span >
303+ < span class ="sd "> display_progress (bool, optional): Whether to display a progress bar. Defaults to True.</ span >
296304
297305< span class ="sd "> Returns:</ span >
298306< span class ="sd "> Tuple[np.ndarray, np.ndarray, np.ndarray]: A tuple containing:</ span >
@@ -338,6 +346,7 @@ <h1>Source code for dte_adj.local</h1><div class="highlight"><pre>
338346 < span class ="n "> control_treatment_arm</ span > < span class ="p "> ,</ span >
339347 < span class ="n "> locations</ span > < span class ="p "> ,</ span >
340348 < span class ="n "> alpha</ span > < span class ="p "> ,</ span >
349+ < span class ="n "> display_progress</ span > < span class ="p "> ,</ span >
341350 < span class ="p "> )</ span > </ div >
342351
343352
@@ -349,6 +358,7 @@ <h1>Source code for dte_adj.local</h1><div class="highlight"><pre>
349358 < span class ="n "> control_treatment_arm</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span >
350359 < span class ="n "> locations</ span > < span class ="p "> :</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span >
351360 < span class ="n "> alpha</ span > < span class ="p "> :</ span > < span class ="nb "> float</ span > < span class ="o "> =</ span > < span class ="mf "> 0.05</ span > < span class ="p "> ,</ span >
361+ < span class ="n "> display_progress</ span > < span class ="p "> :</ span > < span class ="nb "> bool</ span > < span class ="o "> =</ span > < span class ="kc "> True</ span > < span class ="p "> ,</ span >
352362 < span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="n "> Tuple</ span > < span class ="p "> [</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ,</ span > < span class ="n "> np</ span > < span class ="o "> .</ span > < span class ="n "> ndarray</ span > < span class ="p "> ]:</ span >
353363< span class ="w "> </ span > < span class ="sd "> """</ span >
354364< span class ="sd "> Compute Local Probability Treatment Effects (LPTE) using ML adjustment.</ span >
@@ -362,6 +372,7 @@ <h1>Source code for dte_adj.local</h1><div class="highlight"><pre>
362372< span class ="sd "> locations (np.ndarray): Scalar values defining interval boundaries for probability computation.</ span >
363373< span class ="sd "> For each interval (locations[i], locations[i+1]], the LPTE is computed.</ span >
364374< span class ="sd "> alpha (float, optional): Significance level of the confidence bound. Defaults to 0.05.</ span >
375+ < span class ="sd "> display_progress (bool, optional): Whether to display a progress bar. Defaults to True.</ span >
365376
366377< span class ="sd "> Returns:</ span >
367378< span class ="sd "> Tuple[np.ndarray, np.ndarray, np.ndarray]: A tuple containing:</ span >
@@ -410,6 +421,7 @@ <h1>Source code for dte_adj.local</h1><div class="highlight"><pre>
410421 < span class ="n "> control_treatment_arm</ span > < span class ="p "> ,</ span >
411422 < span class ="n "> locations</ span > < span class ="p "> ,</ span >
412423 < span class ="n "> alpha</ span > < span class ="p "> ,</ span >
424+ < span class ="n "> display_progress</ span > < span class ="p "> ,</ span >
413425 < span class ="p "> )</ span > </ div >
414426</ div >
415427
0 commit comments