Merge pull request #88 from pymc-labs/fix-sc-notebooks

Fix synthetic control notebooks

Author: drbenvincent

docs/notebooks/sc2_pymc.ipynb

@@ -15,8 +15,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import pandas as pd\n",
-    "import pathlib\n",
+    "import causalpy as cp\n",
     "import arviz as az"
    ]
   },
@@ -40,12 +39,105 @@
    "cell_type": "code",
    "execution_count": 3,
    "metadata": {},
-   "outputs": [],
+   "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>month</th>\n",
+       "      <th>year</th>\n",
+       "      <th>t</th>\n",
+       "      <th>y</th>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>date</th>\n",
+       "      <th></th>\n",
+       "      <th></th>\n",
+       "      <th></th>\n",
+       "      <th></th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>2010-01-31</th>\n",
+       "      <td>1</td>\n",
+       "      <td>2010</td>\n",
+       "      <td>0</td>\n",
+       "      <td>25.058186</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>2010-02-28</th>\n",
+       "      <td>2</td>\n",
+       "      <td>2010</td>\n",
+       "      <td>1</td>\n",
+       "      <td>27.189812</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>2010-03-31</th>\n",
+       "      <td>3</td>\n",
+       "      <td>2010</td>\n",
+       "      <td>2</td>\n",
+       "      <td>26.487551</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>2010-04-30</th>\n",
+       "      <td>4</td>\n",
+       "      <td>2010</td>\n",
+       "      <td>3</td>\n",
+       "      <td>31.241716</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>2010-05-31</th>\n",
+       "      <td>5</td>\n",
+       "      <td>2010</td>\n",
+       "      <td>4</td>\n",
+       "      <td>40.753973</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "            month  year  t          y\n",
+       "date                                 \n",
+       "2010-01-31      1  2010  0  25.058186\n",
+       "2010-02-28      2  2010  1  27.189812\n",
+       "2010-03-31      3  2010  2  26.487551\n",
+       "2010-04-30      4  2010  3  31.241716\n",
+       "2010-05-31      5  2010  4  40.753973"
+      ]
+     },
+     "execution_count": 3,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
    "source": [
-    "its_data_path = pathlib.Path.cwd().parents[1] / \"causalpy\" / \"data\" / \"its.csv\"\n",
-    "df = pd.read_csv(its_data_path, parse_dates=[\"date\"])\n",
+    "import pandas as pd\n",
+    "\n",
+    "\n",
+    "df = cp.load_data(\"its\")\n",
+    "df[\"date\"] = pd.to_datetime(df[\"date\"])\n",
     "df.set_index(\"date\", inplace=True)\n",
-    "treatment_time = pd.to_datetime(\"2017-01-01\")"
+    "treatment_time = pd.to_datetime(\"2017-01-01\")\n",
+    "df.head()"
    ]
   },
   {
@@ -61,10 +153,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from causalpy.skl_experiments import SyntheticControl\n",
     "from sklearn.linear_model import LinearRegression\n",
     "\n",
-    "result = SyntheticControl(\n",
+    "result = cp.skl_experiments.SyntheticControl(\n",
     "    df,\n",
     "    treatment_time,\n",
     "    formula=\"y ~ 1 + t + C(month)\",\n",