#76 #44 tweaks to DID banks example

Author: drbenvincent

docs/notebooks/did_pymc_banks.ipynb

@@ -6,6 +6,14 @@
    "source": [
     "# Banking dataset with a `pymc` model\n",
     "\n",
+    "<div class=\"alert alert-warning\">\n",
+    "\n",
+    "Warning\n",
+    "\n",
+    "We are still refining the difference in differences analysis code. Expect improvements soon.\n",
+    "\n",
+    "</div>\n",
+    "\n",
     "This notebook analyses historic data on banking closures from [Richardson & Troost (2009)](http://masteringmetrics.com/wp-content/uploads/2015/02/Richardson_Troost_2009_JPE.pdf) and used as a case study for a difference in differences analysis in the [Mastering Metrics](http://www.masteringmetrics.com) book. Here, we replicate this analysis, but using Bayesian inference."
    ]
   },
@@ -650,49 +658,22 @@
    ]
   },
   {
-   "cell_type": "code",
-   "execution_count": null,
+   "cell_type": "markdown",
    "metadata": {},
-   "outputs": [],
    "source": [
-    "from causalpy.pymc_experiments import DifferenceInDifferences\n",
-    "from causalpy.pymc_models import LinearRegression\n",
+    "<div class=\"alert alert-info\">\n",
     "\n",
-    "result = DifferenceInDifferences(\n",
-    "    df_long,\n",
-    "    formula=\"bib ~ 1 + district + year + district:treated\",\n",
-    "    time_variable_name=\"year\",\n",
-    "    prediction_model=LinearRegression(),\n",
-    ")"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "fig, ax = result.plot()"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "result.summary()"
+    "Note\n",
+    "\n",
+    "Coming soon!\n",
+    "\n",
+    "</div>"
    ]
   },
   {
-   "cell_type": "code",
-   "execution_count": null,
+   "cell_type": "markdown",
    "metadata": {},
-   "outputs": [],
-   "source": [
-    "ax = az.plot_posterior(result.causal_impact, ref_val=0)\n",
-    "ax.set(title=\"Posterior estimate of causal impact\");"
-   ]
+   "source": []
   }
  ],
  "metadata": {