Skip to content

Commit

Permalink
Fix subscript ptinting in summary
Browse files Browse the repository at this point in the history
  • Loading branch information
Miruna Oprescu committed Aug 3, 2021
1 parent 28a92b6 commit 2243b26
Show file tree
Hide file tree
Showing 2 changed files with 21 additions and 21 deletions.
2 changes: 1 addition & 1 deletion econml/dml/dynamic_dml.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -682,7 +682,7 @@ def cate_treatment_names(self, treatment_names=None):
slice_treatment_names = super().cate_treatment_names(treatment_names)
treatment_names_out = []
for k in range(self._n_periods):
treatment_names_out += [f"$({t})_{k}$" for t in slice_treatment_names]
treatment_names_out += [f"({t})$_{k}$" for t in slice_treatment_names]
return treatment_names_out

def cate_feature_names(self, feature_names=None):
Expand Down
40 changes: 20 additions & 20 deletions notebooks/Dynamic Double Machine Learning Examples.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -187,7 +187,7 @@
{
"data": {
"text/plain": [
"<econml.dml.dynamic_dml.DynamicDML at 0x24466408a48>"
"<econml.dml.dynamic_dml.DynamicDML at 0x1da18865cc8>"
]
},
"execution_count": 7,
Expand Down Expand Up @@ -281,13 +281,13 @@
" <td></td> <th>point_estimate</th> <th>stderr</th> <th>zstat</th> <th>pvalue</th> <th>ci_lower</th> <th>ci_upper</th>\n",
"</tr>\n",
"<tr>\n",
" <th>cate_intercept|$(T0)_0$</th> <td>0.711</td> <td>0.041</td> <td>17.224</td> <td>0.0</td> <td>0.643</td> <td>0.779</td> \n",
" <th>cate_intercept|(T0)$_0$</th> <td>0.711</td> <td>0.041</td> <td>17.224</td> <td>0.0</td> <td>0.643</td> <td>0.779</td> \n",
"</tr>\n",
"<tr>\n",
" <th>cate_intercept|$(T0)_1$</th> <td>1.031</td> <td>0.096</td> <td>10.715</td> <td>0.0</td> <td>0.872</td> <td>1.189</td> \n",
" <th>cate_intercept|(T0)$_1$</th> <td>1.031</td> <td>0.096</td> <td>10.715</td> <td>0.0</td> <td>0.872</td> <td>1.189</td> \n",
"</tr>\n",
"<tr>\n",
" <th>cate_intercept|$(T0)_2$</th> <td>0.518</td> <td>0.142</td> <td>3.658</td> <td>0.0</td> <td>0.285</td> <td>0.751</td> \n",
" <th>cate_intercept|(T0)$_2$</th> <td>0.518</td> <td>0.142</td> <td>3.658</td> <td>0.0</td> <td>0.285</td> <td>0.751</td> \n",
"</tr>\n",
"</table><br/><br/><sub>A linear parametric conditional average treatment effect (CATE) model was fitted:<br/>$Y = \\Theta(X)\\cdot T + g(X, W) + \\epsilon$<br/>where for every outcome $i$ and treatment $j$ the CATE $\\Theta_{ij}(X)$ has the form:<br/>$\\Theta_{ij}(X) = \\phi(X)' coef_{ij} + cate\\_intercept_{ij}$<br/>where $\\phi(X)$ is the output of the `featurizer` or $X$ if `featurizer`=None. Coefficient Results table portrays the $coef_{ij}$ parameter vector for each outcome $i$ and treatment $j$. Intercept Results table portrays the $cate\\_intercept_{ij}$ parameter.</sub>"
],
Expand All @@ -298,9 +298,9 @@
"=============================================================================\n",
" point_estimate stderr zstat pvalue ci_lower ci_upper\n",
"-----------------------------------------------------------------------------\n",
"cate_intercept|$(T0)_0$ 0.711 0.041 17.224 0.0 0.643 0.779\n",
"cate_intercept|$(T0)_1$ 1.031 0.096 10.715 0.0 0.872 1.189\n",
"cate_intercept|$(T0)_2$ 0.518 0.142 3.658 0.0 0.285 0.751\n",
"cate_intercept|(T0)$_0$ 0.711 0.041 17.224 0.0 0.643 0.779\n",
"cate_intercept|(T0)$_1$ 1.031 0.096 10.715 0.0 0.872 1.189\n",
"cate_intercept|(T0)$_2$ 0.518 0.142 3.658 0.0 0.285 0.751\n",
"-----------------------------------------------------------------------------\n",
"\n",
"<sub>A linear parametric conditional average treatment effect (CATE) model was fitted:\n",
Expand Down Expand Up @@ -446,7 +446,7 @@
{
"data": {
"text/plain": [
"<econml.dml.dynamic_dml.DynamicDML at 0x2446689cd08>"
"<econml.dml.dynamic_dml.DynamicDML at 0x1da188e8a08>"
]
},
"execution_count": 17,
Expand All @@ -472,13 +472,13 @@
" <td></td> <th>point_estimate</th> <th>stderr</th> <th>zstat</th> <th>pvalue</th> <th>ci_lower</th> <th>ci_upper</th>\n",
"</tr>\n",
"<tr>\n",
" <th>X0|$(T0)_0$</th> <td>0.394</td> <td>0.103</td> <td>3.838</td> <td>0.0</td> <td>0.225</td> <td>0.563</td> \n",
" <th>X0|(T0)$_0$</th> <td>0.394</td> <td>0.103</td> <td>3.838</td> <td>0.0</td> <td>0.225</td> <td>0.563</td> \n",
"</tr>\n",
"<tr>\n",
" <th>X0|$(T0)_1$</th> <td>-0.066</td> <td>0.191</td> <td>-0.345</td> <td>0.73</td> <td>-0.38</td> <td>0.248</td> \n",
" <th>X0|(T0)$_1$</th> <td>-0.066</td> <td>0.191</td> <td>-0.345</td> <td>0.73</td> <td>-0.38</td> <td>0.248</td> \n",
"</tr>\n",
"<tr>\n",
" <th>X0|$(T0)_2$</th> <td>0.04</td> <td>0.2</td> <td>0.199</td> <td>0.843</td> <td>-0.29</td> <td>0.369</td> \n",
" <th>X0|(T0)$_2$</th> <td>0.04</td> <td>0.2</td> <td>0.199</td> <td>0.843</td> <td>-0.29</td> <td>0.369</td> \n",
"</tr>\n",
"</table>\n",
"<table class=\"simpletable\">\n",
Expand All @@ -487,13 +487,13 @@
" <td></td> <th>point_estimate</th> <th>stderr</th> <th>zstat</th> <th>pvalue</th> <th>ci_lower</th> <th>ci_upper</th>\n",
"</tr>\n",
"<tr>\n",
" <th>cate_intercept|$(T0)_0$</th> <td>0.579</td> <td>0.052</td> <td>11.242</td> <td>0.0</td> <td>0.495</td> <td>0.664</td> \n",
" <th>cate_intercept|(T0)$_0$</th> <td>0.579</td> <td>0.052</td> <td>11.242</td> <td>0.0</td> <td>0.495</td> <td>0.664</td> \n",
"</tr>\n",
"<tr>\n",
" <th>cate_intercept|$(T0)_1$</th> <td>0.032</td> <td>0.086</td> <td>0.379</td> <td>0.705</td> <td>-0.108</td> <td>0.173</td> \n",
" <th>cate_intercept|(T0)$_1$</th> <td>0.032</td> <td>0.086</td> <td>0.379</td> <td>0.705</td> <td>-0.108</td> <td>0.173</td> \n",
"</tr>\n",
"<tr>\n",
" <th>cate_intercept|$(T0)_2$</th> <td>-0.098</td> <td>0.093</td> <td>-1.048</td> <td>0.294</td> <td>-0.251</td> <td>0.056</td> \n",
" <th>cate_intercept|(T0)$_2$</th> <td>-0.098</td> <td>0.093</td> <td>-1.048</td> <td>0.294</td> <td>-0.251</td> <td>0.056</td> \n",
"</tr>\n",
"</table><br/><br/><sub>A linear parametric conditional average treatment effect (CATE) model was fitted:<br/>$Y = \\Theta(X)\\cdot T + g(X, W) + \\epsilon$<br/>where for every outcome $i$ and treatment $j$ the CATE $\\Theta_{ij}(X)$ has the form:<br/>$\\Theta_{ij}(X) = \\phi(X)' coef_{ij} + cate\\_intercept_{ij}$<br/>where $\\phi(X)$ is the output of the `featurizer` or $X$ if `featurizer`=None. Coefficient Results table portrays the $coef_{ij}$ parameter vector for each outcome $i$ and treatment $j$. Intercept Results table portrays the $cate\\_intercept_{ij}$ parameter.</sub>"
],
Expand All @@ -504,16 +504,16 @@
"=================================================================\n",
" point_estimate stderr zstat pvalue ci_lower ci_upper\n",
"-----------------------------------------------------------------\n",
"X0|$(T0)_0$ 0.394 0.103 3.838 0.0 0.225 0.563\n",
"X0|$(T0)_1$ -0.066 0.191 -0.345 0.73 -0.38 0.248\n",
"X0|$(T0)_2$ 0.04 0.2 0.199 0.843 -0.29 0.369\n",
"X0|(T0)$_0$ 0.394 0.103 3.838 0.0 0.225 0.563\n",
"X0|(T0)$_1$ -0.066 0.191 -0.345 0.73 -0.38 0.248\n",
"X0|(T0)$_2$ 0.04 0.2 0.199 0.843 -0.29 0.369\n",
" CATE Intercept Results \n",
"=============================================================================\n",
" point_estimate stderr zstat pvalue ci_lower ci_upper\n",
"-----------------------------------------------------------------------------\n",
"cate_intercept|$(T0)_0$ 0.579 0.052 11.242 0.0 0.495 0.664\n",
"cate_intercept|$(T0)_1$ 0.032 0.086 0.379 0.705 -0.108 0.173\n",
"cate_intercept|$(T0)_2$ -0.098 0.093 -1.048 0.294 -0.251 0.056\n",
"cate_intercept|(T0)$_0$ 0.579 0.052 11.242 0.0 0.495 0.664\n",
"cate_intercept|(T0)$_1$ 0.032 0.086 0.379 0.705 -0.108 0.173\n",
"cate_intercept|(T0)$_2$ -0.098 0.093 -1.048 0.294 -0.251 0.056\n",
"-----------------------------------------------------------------------------\n",
"\n",
"<sub>A linear parametric conditional average treatment effect (CATE) model was fitted:\n",
Expand Down

0 comments on commit 2243b26

Please sign in to comment.