diff --git a/examples/causal_inference/bayesian_nonparametric_causal.ipynb b/examples/causal_inference/bayesian_nonparametric_causal.ipynb index 1ad4e2e28..aabcbe398 100644 --- a/examples/causal_inference/bayesian_nonparametric_causal.ipynb +++ b/examples/causal_inference/bayesian_nonparametric_causal.ipynb @@ -495,844 +495,844 @@ "data": { "text/html": [ "\n", - "
| \n", - " | trt | \n", - "sex | \n", - "race | \n", - "active_1 | \n", - "active_2 | \n", - "education_2 | \n", - "outcome count | \n", - "outcome mean | \n", - "global_avg | \n", - "diff | \n", + "trt | \n", + "sex | \n", + "race | \n", + "active_1 | \n", + "active_2 | \n", + "education_2 | \n", + "outcome count | \n", + "outcome mean | \n", + "global_avg | \n", + "diff | \n", "|
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 0 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "193 | \n", - "2.858158 | \n", - "2.638300 | \n", - "0.219859 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "193 | \n", + "2.858158 | \n", + "2.638300 | \n", + "0.219859 | \n", "
| 1 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "46 | \n", - "3.870131 | \n", - "2.638300 | \n", - "1.231831 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "46 | \n", + "3.870131 | \n", + "2.638300 | \n", + "1.231831 | \n", "
| 2 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "29 | \n", - "4.095394 | \n", - "2.638300 | \n", - "1.457095 | \n", + "2 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "29 | \n", + "4.095394 | \n", + "2.638300 | \n", + "1.457095 | \n", "
| 3 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "5 | \n", - "0.568137 | \n", - "2.638300 | \n", - "-2.070163 | \n", + "3 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "5 | \n", + "0.568137 | \n", + "2.638300 | \n", + "-2.070163 | \n", "
| 4 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "160 | \n", - "0.709439 | \n", - "2.638300 | \n", - "-1.928861 | \n", + "4 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "160 | \n", + "0.709439 | \n", + "2.638300 | \n", + "-1.928861 | \n", "
| 5 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "36 | \n", - "0.994271 | \n", - "2.638300 | \n", - "-1.644029 | \n", + "5 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "36 | \n", + "0.994271 | \n", + "2.638300 | \n", + "-1.644029 | \n", "
| 6 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "36 | \n", - "2.888559 | \n", - "2.638300 | \n", - "0.250259 | \n", + "6 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "36 | \n", + "2.888559 | \n", + "2.638300 | \n", + "0.250259 | \n", "
| 7 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "4 | \n", - "6.322334 | \n", - "2.638300 | \n", - "3.684034 | \n", + "7 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "4 | \n", + "6.322334 | \n", + "2.638300 | \n", + "3.684034 | \n", "
| 8 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "4 | \n", - "-5.501240 | \n", - "2.638300 | \n", - "-8.139540 | \n", + "8 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "4 | \n", + "-5.501240 | \n", + "2.638300 | \n", + "-8.139540 | \n", "
| 9 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "20 | \n", - "-1.354505 | \n", - "2.638300 | \n", - "-3.992804 | \n", + "9 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "20 | \n", + "-1.354505 | \n", + "2.638300 | \n", + "-3.992804 | \n", "
| 10 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "9 | \n", - "0.442138 | \n", - "2.638300 | \n", - "-2.196162 | \n", + "10 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "9 | \n", + "0.442138 | \n", + "2.638300 | \n", + "-2.196162 | \n", "
| 11 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "157 | \n", - "2.732690 | \n", - "2.638300 | \n", - "0.094390 | \n", + "11 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "157 | \n", + "2.732690 | \n", + "2.638300 | \n", + "0.094390 | \n", "
| 12 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "59 | \n", - "2.222754 | \n", - "2.638300 | \n", - "-0.415546 | \n", + "12 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "59 | \n", + "2.222754 | \n", + "2.638300 | \n", + "-0.415546 | \n", "
| 13 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "36 | \n", - "2.977257 | \n", - "2.638300 | \n", - "0.338957 | \n", + "13 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "36 | \n", + "2.977257 | \n", + "2.638300 | \n", + "0.338957 | \n", "
| 14 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "17 | \n", - "2.087297 | \n", - "2.638300 | \n", - "-0.551003 | \n", + "14 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "17 | \n", + "2.087297 | \n", + "2.638300 | \n", + "-0.551003 | \n", "
| 15 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "200 | \n", - "1.700405 | \n", - "2.638300 | \n", - "-0.937895 | \n", + "15 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "200 | \n", + "1.700405 | \n", + "2.638300 | \n", + "-0.937895 | \n", "
| 16 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "55 | \n", - "-0.492455 | \n", - "2.638300 | \n", - "-3.130754 | \n", + "16 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "55 | \n", + "-0.492455 | \n", + "2.638300 | \n", + "-3.130754 | \n", "
| 17 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "19 | \n", - "2.644629 | \n", - "2.638300 | \n", - "0.006329 | \n", + "17 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "19 | \n", + "2.644629 | \n", + "2.638300 | \n", + "0.006329 | \n", "
| 18 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "18 | \n", - "3.047791 | \n", - "2.638300 | \n", - "0.409491 | \n", + "18 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "18 | \n", + "3.047791 | \n", + "2.638300 | \n", + "0.409491 | \n", "
| 19 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "9 | \n", - "1.637378 | \n", - "2.638300 | \n", - "-1.000922 | \n", + "19 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "9 | \n", + "1.637378 | \n", + "2.638300 | \n", + "-1.000922 | \n", "
| 20 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "4 | \n", - "0.735846 | \n", - "2.638300 | \n", - "-1.902454 | \n", + "20 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "4 | \n", + "0.735846 | \n", + "2.638300 | \n", + "-1.902454 | \n", "
| 21 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "34 | \n", - "0.647564 | \n", - "2.638300 | \n", - "-1.990736 | \n", + "21 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "34 | \n", + "0.647564 | \n", + "2.638300 | \n", + "-1.990736 | \n", "
| 22 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "13 | \n", - "4.815856 | \n", - "2.638300 | \n", - "2.177556 | \n", + "22 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "13 | \n", + "4.815856 | \n", + "2.638300 | \n", + "2.177556 | \n", "
| 23 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "76 | \n", - "4.737206 | \n", - "2.638300 | \n", - "2.098906 | \n", + "23 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "76 | \n", + "4.737206 | \n", + "2.638300 | \n", + "2.098906 | \n", "
| 24 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "18 | \n", - "5.242349 | \n", - "2.638300 | \n", - "2.604049 | \n", + "24 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "18 | \n", + "5.242349 | \n", + "2.638300 | \n", + "2.604049 | \n", "
| 25 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "23 | \n", - "3.205170 | \n", - "2.638300 | \n", - "0.566870 | \n", + "25 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "23 | \n", + "3.205170 | \n", + "2.638300 | \n", + "0.566870 | \n", "
| 26 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "4 | \n", - "6.067620 | \n", - "2.638300 | \n", - "3.429320 | \n", + "26 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "4 | \n", + "6.067620 | \n", + "2.638300 | \n", + "3.429320 | \n", "
| 27 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "70 | \n", - "4.630845 | \n", - "2.638300 | \n", - "1.992545 | \n", + "27 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "70 | \n", + "4.630845 | \n", + "2.638300 | \n", + "1.992545 | \n", "
| 28 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "12 | \n", - "7.570608 | \n", - "2.638300 | \n", - "4.932308 | \n", + "28 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "12 | \n", + "7.570608 | \n", + "2.638300 | \n", + "4.932308 | \n", "
| 29 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "4 | \n", - "7.201967 | \n", - "2.638300 | \n", - "4.563668 | \n", + "29 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "4 | \n", + "7.201967 | \n", + "2.638300 | \n", + "4.563668 | \n", "
| 30 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "3 | \n", - "10.698826 | \n", - "2.638300 | \n", - "8.060526 | \n", + "30 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "3 | \n", + "10.698826 | \n", + "2.638300 | \n", + "8.060526 | \n", "
| 31 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "7 | \n", - "0.778359 | \n", - "2.638300 | \n", - "-1.859941 | \n", + "31 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "7 | \n", + "0.778359 | \n", + "2.638300 | \n", + "-1.859941 | \n", "
| 32 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "3 | \n", - "9.790449 | \n", - "2.638300 | \n", - "7.152149 | \n", + "32 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "3 | \n", + "9.790449 | \n", + "2.638300 | \n", + "7.152149 | \n", "
| 33 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "55 | \n", - "5.095007 | \n", - "2.638300 | \n", - "2.456708 | \n", + "33 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "55 | \n", + "5.095007 | \n", + "2.638300 | \n", + "2.456708 | \n", "
| 34 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "7 | \n", - "9.832617 | \n", - "2.638300 | \n", - "7.194318 | \n", + "34 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "7 | \n", + "9.832617 | \n", + "2.638300 | \n", + "7.194318 | \n", "
| 35 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "14 | \n", - "-1.587808 | \n", - "2.638300 | \n", - "-4.226108 | \n", + "35 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "14 | \n", + "-1.587808 | \n", + "2.638300 | \n", + "-4.226108 | \n", "
| 36 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "1 | \n", - "4 | \n", - "8.761674 | \n", - "2.638300 | \n", - "6.123375 | \n", + "36 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "1 | \n", + "4 | \n", + "8.761674 | \n", + "2.638300 | \n", + "6.123375 | \n", "
| 37 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "67 | \n", - "3.862593 | \n", - "2.638300 | \n", - "1.224293 | \n", + "37 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "67 | \n", + "3.862593 | \n", + "2.638300 | \n", + "1.224293 | \n", "
| 38 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "17 | \n", - "3.162162 | \n", - "2.638300 | \n", - "0.523862 | \n", + "38 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "17 | \n", + "3.162162 | \n", + "2.638300 | \n", + "0.523862 | \n", "
| 39 | \n", - "1 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "0 | \n", - "5 | \n", - "0.522196 | \n", - "2.638300 | \n", - "-2.116104 | \n", + "39 | \n", + "1 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "0 | \n", + "5 | \n", + "0.522196 | \n", + "2.638300 | \n", + "-2.116104 | \n", "
| 40 | \n", - "1 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "1 | \n", - "2 | \n", - "7.826238 | \n", - "2.638300 | \n", - "5.187938 | \n", + "40 | \n", + "1 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "1 | \n", + "2 | \n", + "7.826238 | \n", + "2.638300 | \n", + "5.187938 | \n", "
| 41 | \n", - "1 | \n", - "1 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "0 | \n", - "8 | \n", - "5.756044 | \n", - "2.638300 | \n", - "3.117744 | \n", + "41 | \n", + "1 | \n", + "1 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "0 | \n", + "8 | \n", + "5.756044 | \n", + "2.638300 | \n", + "3.117744 | \n", "
| 42 | \n", - "1 | \n", - "1 | \n", - "1 | \n", - "1 | \n", - "0 | \n", - "1 | \n", - "4 | \n", - "5.440875 | \n", - "2.638300 | \n", - "2.802575 | \n", + "42 | \n", + "1 | \n", + "1 | \n", + "1 | \n", + "1 | \n", + "0 | \n", + "1 | \n", + "4 | \n", + "5.440875 | \n", + "2.638300 | \n", + "2.802575 | \n", "
<xarray.DataArray 'p' (p_dim_0: 1566)>\n",
"array([0.1 , 0.15, 0.13, ..., 0.13, 0.47, 0.18])\n",
"Coordinates:\n",
- " * p_dim_0 (p_dim_0) int64 0 1 2 3 4 5 6 ... 1560 1561 1562 1563 1564 1565"
+ " * p_dim_0 (p_dim_0) int64 0 1 2 3 4 5 6 ... 1560 1561 1562 1563 1564 1565"
],
"text/plain": [
"<xarray.DataArray 'p' (p_dim_0: 1566)>\n",
- "array([0.18, 0.18, 0.17, ..., 0.16, 0.32, 0.28])\n",
+ "array([0.19, 0.18, 0.23, ..., 0.18, 0.31, 0.21])\n",
"Coordinates:\n",
- " * p_dim_0 (p_dim_0) int64 0 1 2 3 4 5 6 ... 1560 1561 1562 1563 1564 1565"
+ " * p_dim_0 (p_dim_0) int64 0 1 2 3 4 5 6 ... 1560 1561 1562 1563 1564 1565"
],
"text/plain": [
"PandasIndex(Index([0, 1, 2, 3], dtype='int64', name='chain'))
PandasIndex(Index([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,\n", + " ...\n", + " 990, 991, 992, 993, 994, 995, 996, 997, 998, 999],\n", + " dtype='int64', name='draw', length=1000))
PandasIndex(Index([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,\n", + " ...\n", + " 1556, 1557, 1558, 1559, 1560, 1561, 1562, 1563, 1564, 1565],\n", + " dtype='int64', name='obs', length=1566))
PandasIndex(Index([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,\n", + " ...\n", + " 1556, 1557, 1558, 1559, 1560, 1561, 1562, 1563, 1564, 1565],\n", + " dtype='int64', name='X_dim_0', length=1566))
PandasIndex(Index([0, 1, 2], dtype='int64', name='X_dim_1'))
PandasIndex(Index([0, 1, 2, 3], dtype='int64', name='chain'))
PandasIndex(Index([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,\n", + " ...\n", + " 990, 991, 992, 993, 994, 995, 996, 997, 998, 999],\n", + " dtype='int64', name='draw', length=1000))
PandasIndex(Index([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,\n", + " ...\n", + " 1556, 1557, 1558, 1559, 1560, 1561, 1562, 1563, 1564, 1565],\n", + " dtype='int64', name='obs', length=1566))
PandasIndex(Index([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9,\n", + " ...\n", + " 1556, 1557, 1558, 1559, 1560, 1561, 1562, 1563, 1564, 1565],\n", + " dtype='int64', name='X_dim_0', length=1566))
PandasIndex(Index([0, 1, 2], dtype='int64', name='X_dim_1'))
<xarray.DataArray 'pred' ()>\n", - "array(3.91691031)" + "array(3.91927737)" ], "text/plain": [ "
<xarray.DataArray 'pred' ()>\n", - "array(0.82496553)" + "array(0.78187425)" ], "text/plain": [ "
<xarray.DataArray 'pred' ()>\n", - "array(3.09194478)" + "array(3.13740312)" ], "text/plain": [ "
| \n", - " | smoke | \n", - "sex | \n", - "race | \n", - "phealth | \n", - "log_y count | \n", - "log_y mean | \n", - "log_y std | \n", - "global_avg | \n", - "diff | \n", + "smoke | \n", + "sex | \n", + "race | \n", + "phealth | \n", + "log_y count | \n", + "log_y mean | \n", + "log_y std | \n", + "global_avg | \n", + "diff | \n", "|
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 29 | \n", - "0 | \n", - "Female | \n", - "White | \n", - "Very Good | \n", - "1858 | \n", - "8.101406 | \n", - "0.896128 | \n", - "8.089203 | \n", - "0.012204 | \n", + "29 | \n", + "0 | \n", + "Female | \n", + "White | \n", + "Very Good | \n", + "1858 | \n", + "8.101406 | \n", + "0.896128 | \n", + "8.089203 | \n", + "0.012204 | \n", "
| 27 | \n", - "0 | \n", - "Female | \n", - "White | \n", - "Good | \n", - "1572 | \n", - "8.231117 | \n", - "1.010783 | \n", - "8.089203 | \n", - "0.141914 | \n", + "27 | \n", + "0 | \n", + "Female | \n", + "White | \n", + "Good | \n", + "1572 | \n", + "8.231117 | \n", + "1.010783 | \n", + "8.089203 | \n", + "0.141914 | \n", "
| 25 | \n", - "0 | \n", - "Female | \n", - "White | \n", - "Excellent | \n", - "1385 | \n", - "7.919802 | \n", - "0.846725 | \n", - "8.089203 | \n", - "-0.169400 | \n", + "25 | \n", + "0 | \n", + "Female | \n", + "White | \n", + "Excellent | \n", + "1385 | \n", + "7.919802 | \n", + "0.846725 | \n", + "8.089203 | \n", + "-0.169400 | \n", "
| 59 | \n", - "0 | \n", - "Male | \n", - "White | \n", - "Very Good | \n", - "1321 | \n", - "7.987652 | \n", - "0.922520 | \n", - "8.089203 | \n", - "-0.101551 | \n", + "59 | \n", + "0 | \n", + "Male | \n", + "White | \n", + "Very Good | \n", + "1321 | \n", + "7.987652 | \n", + "0.922520 | \n", + "8.089203 | \n", + "-0.101551 | \n", "
| 57 | \n", - "0 | \n", - "Male | \n", - "White | \n", - "Good | \n", - "1129 | \n", - "8.178290 | \n", - "1.003363 | \n", - "8.089203 | \n", - "0.089088 | \n", + "57 | \n", + "0 | \n", + "Male | \n", + "White | \n", + "Good | \n", + "1129 | \n", + "8.178290 | \n", + "1.003363 | \n", + "8.089203 | \n", + "0.089088 | \n", "
| 55 | \n", - "0 | \n", - "Male | \n", - "White | \n", - "Excellent | \n", - "1122 | \n", - "7.728966 | \n", - "0.779346 | \n", - "8.089203 | \n", - "-0.360236 | \n", + "55 | \n", + "0 | \n", + "Male | \n", + "White | \n", + "Excellent | \n", + "1122 | \n", + "7.728966 | \n", + "0.779346 | \n", + "8.089203 | \n", + "-0.360236 | \n", "
| 26 | \n", - "0 | \n", - "Female | \n", - "White | \n", - "Fair | \n", - "659 | \n", - "8.487774 | \n", - "1.113656 | \n", - "8.089203 | \n", - "0.398572 | \n", + "26 | \n", + "0 | \n", + "Female | \n", + "White | \n", + "Fair | \n", + "659 | \n", + "8.487774 | \n", + "1.113656 | \n", + "8.089203 | \n", + "0.398572 | \n", "
| 7 | \n", - "0 | \n", - "Female | \n", - "Black | \n", - "Good | \n", - "515 | \n", - "8.125243 | \n", - "0.944796 | \n", - "8.089203 | \n", - "0.036040 | \n", + "7 | \n", + "0 | \n", + "Female | \n", + "Black | \n", + "Good | \n", + "515 | \n", + "8.125243 | \n", + "0.944796 | \n", + "8.089203 | \n", + "0.036040 | \n", "
| 9 | \n", - "0 | \n", - "Female | \n", - "Black | \n", - "Very Good | \n", - "488 | \n", - "7.870293 | \n", - "0.884956 | \n", - "8.089203 | \n", - "-0.218909 | \n", + "9 | \n", + "0 | \n", + "Female | \n", + "Black | \n", + "Very Good | \n", + "488 | \n", + "7.870293 | \n", + "0.884956 | \n", + "8.089203 | \n", + "-0.218909 | \n", "
| 56 | \n", - "0 | \n", - "Male | \n", - "White | \n", - "Fair | \n", - "434 | \n", - "8.601018 | \n", - "1.112748 | \n", - "8.089203 | \n", - "0.511816 | \n", + "56 | \n", + "0 | \n", + "Male | \n", + "White | \n", + "Fair | \n", + "434 | \n", + "8.601018 | \n", + "1.112748 | \n", + "8.089203 | \n", + "0.511816 | \n", "
| 110 | \n", - "1 | \n", - "Male | \n", - "White | \n", - "Good | \n", - "335 | \n", - "7.939632 | \n", - "0.887826 | \n", - "8.089203 | \n", - "-0.149571 | \n", + "110 | \n", + "1 | \n", + "Male | \n", + "White | \n", + "Good | \n", + "335 | \n", + "7.939632 | \n", + "0.887826 | \n", + "8.089203 | \n", + "-0.149571 | \n", "
| 84 | \n", - "1 | \n", - "Female | \n", - "White | \n", - "Good | \n", - "324 | \n", - "8.077777 | \n", - "0.968686 | \n", - "8.089203 | \n", - "-0.011426 | \n", + "84 | \n", + "1 | \n", + "Female | \n", + "White | \n", + "Good | \n", + "324 | \n", + "8.077777 | \n", + "0.968686 | \n", + "8.089203 | \n", + "-0.011426 | \n", "
| 5 | \n", - "0 | \n", - "Female | \n", - "Black | \n", - "Excellent | \n", - "307 | \n", - "7.748597 | \n", - "0.812461 | \n", - "8.089203 | \n", - "-0.340606 | \n", + "5 | \n", + "0 | \n", + "Female | \n", + "Black | \n", + "Excellent | \n", + "307 | \n", + "7.748597 | \n", + "0.812461 | \n", + "8.089203 | \n", + "-0.340606 | \n", "
| 6 | \n", - "0 | \n", - "Female | \n", - "Black | \n", - "Fair | \n", - "266 | \n", - "8.534893 | \n", - "1.057159 | \n", - "8.089203 | \n", - "0.445690 | \n", + "6 | \n", + "0 | \n", + "Female | \n", + "Black | \n", + "Fair | \n", + "266 | \n", + "8.534893 | \n", + "1.057159 | \n", + "8.089203 | \n", + "0.445690 | \n", "
| 86 | \n", - "1 | \n", - "Female | \n", - "White | \n", - "Very Good | \n", - "266 | \n", - "7.913179 | \n", - "0.902211 | \n", - "8.089203 | \n", - "-0.176024 | \n", + "86 | \n", + "1 | \n", + "Female | \n", + "White | \n", + "Very Good | \n", + "266 | \n", + "7.913179 | \n", + "0.902211 | \n", + "8.089203 | \n", + "-0.176024 | \n", "
| 39 | \n", - "0 | \n", - "Male | \n", - "Black | \n", - "Very Good | \n", - "246 | \n", - "7.765843 | \n", - "0.831623 | \n", - "8.089203 | \n", - "-0.323360 | \n", + "39 | \n", + "0 | \n", + "Male | \n", + "Black | \n", + "Very Good | \n", + "246 | \n", + "7.765843 | \n", + "0.831623 | \n", + "8.089203 | \n", + "-0.323360 | \n", "
| 37 | \n", - "0 | \n", - "Male | \n", - "Black | \n", - "Good | \n", - "235 | \n", - "8.002760 | \n", - "1.051284 | \n", - "8.089203 | \n", - "-0.086443 | \n", + "37 | \n", + "0 | \n", + "Male | \n", + "Black | \n", + "Good | \n", + "235 | \n", + "8.002760 | \n", + "1.051284 | \n", + "8.089203 | \n", + "-0.086443 | \n", "
| 112 | \n", - "1 | \n", - "Male | \n", - "White | \n", - "Very Good | \n", - "235 | \n", - "7.848349 | \n", - "0.900002 | \n", - "8.089203 | \n", - "-0.240854 | \n", + "112 | \n", + "1 | \n", + "Male | \n", + "White | \n", + "Very Good | \n", + "235 | \n", + "7.848349 | \n", + "0.900002 | \n", + "8.089203 | \n", + "-0.240854 | \n", "
| 4 | \n", - "0 | \n", - "Female | \n", - "Asian | \n", - "Very Good | \n", - "193 | \n", - "7.864920 | \n", - "0.859187 | \n", - "8.089203 | \n", - "-0.224283 | \n", + "4 | \n", + "0 | \n", + "Female | \n", + "Asian | \n", + "Very Good | \n", + "193 | \n", + "7.864920 | \n", + "0.859187 | \n", + "8.089203 | \n", + "-0.224283 | \n", "
| 83 | \n", - "1 | \n", - "Female | \n", - "White | \n", - "Fair | \n", - "191 | \n", - "8.403307 | \n", - "0.989581 | \n", - "8.089203 | \n", - "0.314105 | \n", + "83 | \n", + "1 | \n", + "Female | \n", + "White | \n", + "Fair | \n", + "191 | \n", + "8.403307 | \n", + "0.989581 | \n", + "8.089203 | \n", + "0.314105 | \n", "
| 28 | \n", - "0 | \n", - "Female | \n", - "White | \n", - "Poor | \n", - "186 | \n", - "9.160054 | \n", - "1.138894 | \n", - "8.089203 | \n", - "1.070852 | \n", + "28 | \n", + "0 | \n", + "Female | \n", + "White | \n", + "Poor | \n", + "186 | \n", + "9.160054 | \n", + "1.138894 | \n", + "8.089203 | \n", + "1.070852 | \n", "
| 35 | \n", - "0 | \n", - "Male | \n", - "Black | \n", - "Excellent | \n", - "184 | \n", - "7.620076 | \n", - "0.771911 | \n", - "8.089203 | \n", - "-0.469127 | \n", + "35 | \n", + "0 | \n", + "Male | \n", + "Black | \n", + "Excellent | \n", + "184 | \n", + "7.620076 | \n", + "0.771911 | \n", + "8.089203 | \n", + "-0.469127 | \n", "
| 0 | \n", - "0 | \n", - "Female | \n", - "Asian | \n", - "Excellent | \n", - "164 | \n", - "7.786508 | \n", - "0.899504 | \n", - "8.089203 | \n", - "-0.302694 | \n", + "0 | \n", + "0 | \n", + "Female | \n", + "Asian | \n", + "Excellent | \n", + "164 | \n", + "7.786508 | \n", + "0.899504 | \n", + "8.089203 | \n", + "-0.302694 | \n", "
| 2 | \n", - "0 | \n", - "Female | \n", - "Asian | \n", - "Good | \n", - "162 | \n", - "7.873122 | \n", - "0.768487 | \n", - "8.089203 | \n", - "-0.216080 | \n", + "2 | \n", + "0 | \n", + "Female | \n", + "Asian | \n", + "Good | \n", + "162 | \n", + "7.873122 | \n", + "0.768487 | \n", + "8.089203 | \n", + "-0.216080 | \n", "
| 82 | \n", - "1 | \n", - "Female | \n", - "White | \n", - "Excellent | \n", - "149 | \n", - "7.860320 | \n", - "0.837483 | \n", - "8.089203 | \n", - "-0.228882 | \n", + "82 | \n", + "1 | \n", + "Female | \n", + "White | \n", + "Excellent | \n", + "149 | \n", + "7.860320 | \n", + "0.837483 | \n", + "8.089203 | \n", + "-0.228882 | \n", "
| 108 | \n", - "1 | \n", - "Male | \n", - "White | \n", - "Excellent | \n", - "148 | \n", - "7.652529 | \n", - "0.717871 | \n", - "8.089203 | \n", - "-0.436674 | \n", + "108 | \n", + "1 | \n", + "Male | \n", + "White | \n", + "Excellent | \n", + "148 | \n", + "7.652529 | \n", + "0.717871 | \n", + "8.089203 | \n", + "-0.436674 | \n", "
| 109 | \n", - "1 | \n", - "Male | \n", - "White | \n", - "Fair | \n", - "148 | \n", - "8.282303 | \n", - "1.111403 | \n", - "8.089203 | \n", - "0.193101 | \n", + "109 | \n", + "1 | \n", + "Male | \n", + "White | \n", + "Fair | \n", + "148 | \n", + "8.282303 | \n", + "1.111403 | \n", + "8.089203 | \n", + "0.193101 | \n", "
| 58 | \n", - "0 | \n", - "Male | \n", - "White | \n", - "Poor | \n", - "140 | \n", - "9.308711 | \n", - "1.255442 | \n", - "8.089203 | \n", - "1.219509 | \n", + "58 | \n", + "0 | \n", + "Male | \n", + "White | \n", + "Poor | \n", + "140 | \n", + "9.308711 | \n", + "1.255442 | \n", + "8.089203 | \n", + "1.219509 | \n", "
| 34 | \n", - "0 | \n", - "Male | \n", - "Asian | \n", - "Very Good | \n", - "140 | \n", - "7.792831 | \n", - "0.772666 | \n", - "8.089203 | \n", - "-0.296371 | \n", + "34 | \n", + "0 | \n", + "Male | \n", + "Asian | \n", + "Very Good | \n", + "140 | \n", + "7.792831 | \n", + "0.772666 | \n", + "8.089203 | \n", + "-0.296371 | \n", "
| 32 | \n", - "0 | \n", - "Male | \n", - "Asian | \n", - "Good | \n", - "134 | \n", - "7.993583 | \n", - "1.123291 | \n", - "8.089203 | \n", - "-0.095620 | \n", + "32 | \n", + "0 | \n", + "Male | \n", + "Asian | \n", + "Good | \n", + "134 | \n", + "7.993583 | \n", + "1.123291 | \n", + "8.089203 | \n", + "-0.095620 | \n", "