add comment about beta parameters for the quadratic example

pymc-labs · Nov 6, 2023 · dca2844 · dca2844
1 parent cc07b94
commit dca2844
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/docs/source/notebooks/rkink_pymc.ipynb b/docs/source/notebooks/rkink_pymc.ipynb
@@ -425,6 +425,8 @@
    ],
    "source": [
     "kink = 0.5\n",
+    "# quadratic function going from 1*x^2 on the left of the kink point, to 1*x^2 -\n",
+    "# 1*(x-kink) - 5*(x-kink)^2 to the right of the kink point\n",
     "beta = [0, 0, 1, -1, -5]\n",
     "df = generate_data(beta, kink, N=200, sigma=sigma)\n",
     "\n",