Skip to content

Commit 8ce77a1

Browse files
committed
add docstring to RegressionKink class
1 parent 7847ede commit 8ce77a1

File tree

2 files changed

+25
-3
lines changed

2 files changed

+25
-3
lines changed

causalpy/pymc_experiments.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -958,6 +958,28 @@ def summary(self) -> None:
958958

959959

960960
class RegressionKink(ExperimentalDesign):
961+
"""
962+
A class to analyse sharp regression kink experiments.
963+
964+
:param data:
965+
A pandas dataframe
966+
:param formula:
967+
A statistical model formula
968+
:param kink_point:
969+
A scalar threshold value at which there is a change in the first derivative of
970+
the assignment function
971+
:param model:
972+
A PyMC model
973+
:param running_variable_name:
974+
The name of the predictor variable that the kink_point is based upon
975+
:param epsilon:
976+
A small scalar value which determines how far above and below the kink point to
977+
evaluate the causal impact.
978+
:param bandwidth:
979+
Data outside of the bandwidth (relative to the discontinuity) is not used to fit
980+
the model.
981+
"""
982+
961983
def __init__(
962984
self,
963985
data: pd.DataFrame,

docs/source/_static/interrogate_badge.svg

Lines changed: 3 additions & 3 deletions
Loading

0 commit comments

Comments
 (0)