Skip to content

Commit

Permalink
add docstring to RegressionKink class
Browse files Browse the repository at this point in the history
  • Loading branch information
drbenvincent committed Nov 2, 2023
1 parent 7847ede commit 8ce77a1
Show file tree
Hide file tree
Showing 2 changed files with 25 additions and 3 deletions.
22 changes: 22 additions & 0 deletions causalpy/pymc_experiments.py
Original file line number Diff line number Diff line change
Expand Up @@ -958,6 +958,28 @@ def summary(self) -> None:


class RegressionKink(ExperimentalDesign):
"""
A class to analyse sharp regression kink experiments.
:param data:
A pandas dataframe
:param formula:
A statistical model formula
:param kink_point:
A scalar threshold value at which there is a change in the first derivative of
the assignment function
:param model:
A PyMC model
:param running_variable_name:
The name of the predictor variable that the kink_point is based upon
:param epsilon:
A small scalar value which determines how far above and below the kink point to
evaluate the causal impact.
:param bandwidth:
Data outside of the bandwidth (relative to the discontinuity) is not used to fit
the model.
"""

def __init__(
self,
data: pd.DataFrame,
Expand Down
6 changes: 3 additions & 3 deletions docs/source/_static/interrogate_badge.svg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit 8ce77a1

Please sign in to comment.