{
"cells": [
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Regression discontinuity with sci-kit learn models"
]
},
{
"cell_type": "code",
"execution_count": 1,
"metadata": {},
"outputs": [],
"source": [
"from sklearn.gaussian_process import GaussianProcessRegressor\n",
"from sklearn.gaussian_process.kernels import ExpSineSquared, WhiteKernel\n",
"from sklearn.linear_model import LinearRegression\n",
"\n",
"import causalpy as cp"
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {},
"outputs": [],
"source": [
"%config InlineBackend.figure_format = 'retina'"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"## Load data"
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"\n",
"\n",
"
\n",
" \n",
" \n",
" | \n",
" x | \n",
" y | \n",
" treated | \n",
"
\n",
" \n",
" \n",
" \n",
" | 0 | \n",
" -0.932739 | \n",
" -0.091919 | \n",
" False | \n",
"
\n",
" \n",
" | 1 | \n",
" -0.930778 | \n",
" -0.382663 | \n",
" False | \n",
"
\n",
" \n",
" | 2 | \n",
" -0.929110 | \n",
" -0.181786 | \n",
" False | \n",
"
\n",
" \n",
" | 3 | \n",
" -0.907419 | \n",
" -0.288245 | \n",
" False | \n",
"
\n",
" \n",
" | 4 | \n",
" -0.882469 | \n",
" -0.420811 | \n",
" False | \n",
"
\n",
" \n",
"
\n",
"