Skip to content

Commit 3efdb69

Browse files
committed
deploy: a6c7c14
1 parent 958f2b7 commit 3efdb69

6 files changed

Lines changed: 74 additions & 24 deletions

File tree

_modules/dte_adj/local.html

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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">-&gt;</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">&quot;&quot;&quot;</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">-&gt;</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">&quot;&quot;&quot;</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">-&gt;</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">&quot;&quot;&quot;</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">-&gt;</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">&quot;&quot;&quot;</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

Comments
 (0)