Skip to content

Commit fe156af

Browse files
committed
rename TimeSeriesExperiment to PrePostFit
1 parent 39714f2 commit fe156af

File tree

5 files changed

+14
-10
lines changed

5 files changed

+14
-10
lines changed

causalpy/pymc_experiments.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,9 @@ def print_coefficients(self):
5353
)
5454

5555

56-
class TimeSeriesExperiment(ExperimentalDesign):
57-
"""A class to analyse time series quasi-experiments"""
56+
class PrePostFit(ExperimentalDesign):
57+
"""A class to analyse quasi-experiments where parameter estimation is based on just
58+
the pre-intervention data."""
5859

5960
def __init__(
6061
self,
@@ -242,8 +243,8 @@ def summary(self):
242243
self.print_coefficients()
243244

244245

245-
class SyntheticControl(TimeSeriesExperiment):
246-
"""A wrapper around the TimeSeriesExperiment class"""
246+
class SyntheticControl(PrePostFit):
247+
"""A wrapper around the PrePostFit class"""
247248

248249
expt_type = "Synthetic Control"
249250

causalpy/skl_experiments.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,10 @@ def __init__(self, model=None, **kwargs):
2020
raise ValueError("fitting_model not set or passed.")
2121

2222

23-
class TimeSeriesExperiment(ExperimentalDesign):
23+
class PrePostFit(ExperimentalDesign):
24+
"""A class to analyse quasi-experiments where parameter estimation is based on just
25+
the pre-intervention data."""
26+
2427
def __init__(
2528
self,
2629
data,
@@ -148,8 +151,8 @@ def plot_coeffs(self):
148151
)
149152

150153

151-
class SyntheticControl(TimeSeriesExperiment):
152-
"""A wrapper around the TimeSeriesExperiment class"""
154+
class SyntheticControl(PrePostFit):
155+
"""A wrapper around the PrePostFit class"""
153156

154157
def plot(self, plot_predictors=False):
155158
"""Plot the results"""

classes.png

283 KB
Loading

docs/source/_static/interrogate_badge.svg

Lines changed: 3 additions & 3 deletions
Loading

packages.png

57 KB
Loading

0 commit comments

Comments
 (0)