update notebooks

Author: drbenvincent

docs/source/notebooks/did_pymc.ipynb

@@ -169,7 +169,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "948c5e2eb1884852866729a222bc9ca7",
+       "model_id": "b37c2ff6780f4b6a9c7806ee9ea25d50",
        "version_major": 2,
        "version_minor": 0
       },
@@ -288,25 +288,79 @@
    "cell_type": "code",
    "execution_count": 8,
    "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>mean</th>\n",
+       "      <th>median</th>\n",
+       "      <th>hdi_lower</th>\n",
+       "      <th>hdi_upper</th>\n",
+       "      <th>p_gt_0</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>treatment_effect</th>\n",
+       "      <td>0.503779</td>\n",
+       "      <td>0.504367</td>\n",
+       "      <td>0.401661</td>\n",
+       "      <td>0.607651</td>\n",
+       "      <td>1.0</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "                      mean    median  hdi_lower  hdi_upper  p_gt_0\n",
+       "treatment_effect  0.503779  0.504367   0.401661   0.607651     1.0"
+      ]
+     },
+     "execution_count": 8,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# Generate effect summary\n",
+    "stats = result.effect_summary()\n",
+    "stats.table"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "                      mean    median  hdi_lower  hdi_upper  p_gt_0\n",
-      "treatment_effect  0.503779  0.504367   0.401661   0.607651     1.0\n",
-      "\n",
       "The average treatment effect was 0.50 (95% HDI [0.40, 0.61]), with a posterior probability of an increase of 1.000.\n"
      ]
     }
    ],
    "source": [
-    "# Generate effect summary\n",
-    "stats = result.effect_summary()\n",
-    "\n",
-    "# Display the summary table\n",
-    "print(stats.table)\n",
-    "print(\"\\n\" + stats.text)"
+    "print(stats.text)"
    ]
   },
   {
@@ -322,21 +376,65 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 10,
    "metadata": {},
    "outputs": [
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "                      mean    median  hdi_lower  hdi_upper  p_two_sided  \\\n",
-      "treatment_effect  0.503779  0.504367   0.401661   0.607651          0.0   \n",
-      "\n",
-      "                  prob_of_effect   p_rope  \n",
-      "treatment_effect             1.0  0.99925  \n",
-      "\n",
-      "The average treatment effect was 0.50 (95% HDI [0.40, 0.61]), with a posterior probability of an effect of 1.000.\n"
-     ]
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>mean</th>\n",
+       "      <th>median</th>\n",
+       "      <th>hdi_lower</th>\n",
+       "      <th>hdi_upper</th>\n",
+       "      <th>p_two_sided</th>\n",
+       "      <th>prob_of_effect</th>\n",
+       "      <th>p_rope</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>treatment_effect</th>\n",
+       "      <td>0.503779</td>\n",
+       "      <td>0.504367</td>\n",
+       "      <td>0.401661</td>\n",
+       "      <td>0.607651</td>\n",
+       "      <td>0.0</td>\n",
+       "      <td>1.0</td>\n",
+       "      <td>0.99925</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "                      mean    median  hdi_lower  hdi_upper  p_two_sided  \\\n",
+       "treatment_effect  0.503779  0.504367   0.401661   0.607651          0.0   \n",
+       "\n",
+       "                  prob_of_effect   p_rope  \n",
+       "treatment_effect             1.0  0.99925  "
+      ]
+     },
+     "execution_count": 10,
+     "metadata": {},
+     "output_type": "execute_result"
     }
    ],
    "source": [
@@ -346,8 +444,24 @@
     "    alpha=0.05,\n",
     "    min_effect=0.3,  # Region of Practical Equivalence\n",
     ")\n",
-    "\n",
-    "print(stats.table)\n",
+    "stats.table"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "The average treatment effect was 0.50 (95% HDI [0.40, 0.61]), with a posterior probability of an effect of 1.000.\n"
+     ]
+    }
+   ],
+   "source": [
     "print(\"\\n\" + stats.text)"
    ]
   },