deploy: d283bf7

Author: TomeHirata

_modules/dte_adj/stratified.html

@@ -110,10 +110,9 @@ <h1>Source code for dte_adj.stratified</h1><div class="highlight"><pre>
         <span class="k">for</span> <span class="n">s</span> <span class="ow">in</span> <span class="n">s_list</span><span class="p">:</span>
             <span class="n">s_mask</span> <span class="o">=</span> <span class="n">strata</span> <span class="o">==</span> <span class="n">s</span>
             <span class="n">w_s</span><span class="p">[</span><span class="n">s</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span><span class="n">s_mask</span> <span class="o">&amp;</span> <span class="n">treatment_mask</span><span class="p">)</span><span class="o">.</span><span class="n">sum</span><span class="p">()</span> <span class="o">/</span> <span class="n">s_mask</span><span class="o">.</span><span class="n">sum</span><span class="p">()</span>
-        <span class="k">for</span> <span class="n">i</span><span class="p">,</span> <span class="n">outcome</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span><span class="n">locations</span><span class="p">):</span>
-            <span class="k">for</span> <span class="n">j</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">n_records</span><span class="p">):</span>
-                <span class="n">s</span> <span class="o">=</span> <span class="n">strata</span><span class="p">[</span><span class="n">j</span><span class="p">]</span>
-                <span class="n">prediction</span><span class="p">[</span><span class="n">j</span><span class="p">,</span> <span class="n">i</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span><span class="n">outcomes</span><span class="p">[</span><span class="n">j</span><span class="p">]</span> <span class="o">&lt;=</span> <span class="n">outcome</span><span class="p">)</span> <span class="o">/</span> <span class="n">w_s</span><span class="p">[</span><span class="n">s</span><span class="p">]</span> <span class="o">*</span> <span class="n">treatment_mask</span><span class="p">[</span><span class="n">j</span><span class="p">]</span>
+        <span class="k">for</span> <span class="n">j</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">n_records</span><span class="p">):</span>
+            <span class="n">s</span> <span class="o">=</span> <span class="n">strata</span><span class="p">[</span><span class="n">j</span><span class="p">]</span>
+            <span class="n">prediction</span><span class="p">[</span><span class="n">j</span><span class="p">]</span> <span class="o">=</span> <span class="p">(</span><span class="n">outcomes</span><span class="p">[</span><span class="n">j</span><span class="p">]</span> <span class="o">&lt;=</span> <span class="n">locations</span><span class="p">)</span> <span class="o">/</span> <span class="n">w_s</span><span class="p">[</span><span class="n">s</span><span class="p">]</span> <span class="o">*</span> <span class="n">treatment_mask</span><span class="p">[</span><span class="n">j</span><span class="p">]</span>
 
         <span class="n">unconditional_pred</span> <span class="o">=</span> <span class="p">{</span><span class="n">s</span><span class="p">:</span> <span class="n">prediction</span><span class="p">[</span><span class="n">s</span> <span class="o">==</span> <span class="n">strata</span><span class="p">]</span><span class="o">.</span><span class="n">mean</span><span class="p">(</span><span class="n">axis</span><span class="o">=</span><span class="mi">0</span><span class="p">)</span> <span class="k">for</span> <span class="n">s</span> <span class="ow">in</span> <span class="n">s_list</span><span class="p">}</span>
         <span class="n">conditional_prediction</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">array</span><span class="p">([</span><span class="n">unconditional_pred</span><span class="p">[</span><span class="n">s</span><span class="p">]</span> <span class="k">for</span> <span class="n">s</span> <span class="ow">in</span> <span class="n">strata</span><span class="p">])</span>