diff-diff 0.6.0__py3-none-any.whl → 1.0.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- diff_diff/__init__.py +35 -1
- diff_diff/bacon.py +1027 -0
- diff_diff/estimators.py +138 -1
- diff_diff/power.py +1350 -0
- diff_diff/staggered.py +564 -13
- diff_diff/visualization.py +568 -0
- {diff_diff-0.6.0.dist-info → diff_diff-1.0.0.dist-info}/METADATA +54 -6
- diff_diff-1.0.0.dist-info/RECORD +15 -0
- diff_diff-0.6.0.dist-info/RECORD +0 -13
- {diff_diff-0.6.0.dist-info → diff_diff-1.0.0.dist-info}/WHEEL +0 -0
- {diff_diff-0.6.0.dist-info → diff_diff-1.0.0.dist-info}/top_level.txt +0 -0
diff_diff/__init__.py
CHANGED
|
@@ -5,6 +5,12 @@ This library provides sklearn-like estimators for causal inference
|
|
|
5
5
|
using the difference-in-differences methodology.
|
|
6
6
|
"""
|
|
7
7
|
|
|
8
|
+
from diff_diff.bacon import (
|
|
9
|
+
BaconDecomposition,
|
|
10
|
+
BaconDecompositionResults,
|
|
11
|
+
Comparison2x2,
|
|
12
|
+
bacon_decompose,
|
|
13
|
+
)
|
|
8
14
|
from diff_diff.diagnostics import (
|
|
9
15
|
PlaceboTestResults,
|
|
10
16
|
leave_one_out_test,
|
|
@@ -30,6 +36,15 @@ from diff_diff.honest_did import (
|
|
|
30
36
|
compute_honest_did,
|
|
31
37
|
sensitivity_plot,
|
|
32
38
|
)
|
|
39
|
+
from diff_diff.power import (
|
|
40
|
+
PowerAnalysis,
|
|
41
|
+
PowerResults,
|
|
42
|
+
SimulationPowerResults,
|
|
43
|
+
compute_mde,
|
|
44
|
+
compute_power,
|
|
45
|
+
compute_sample_size,
|
|
46
|
+
simulate_power,
|
|
47
|
+
)
|
|
33
48
|
from diff_diff.prep import (
|
|
34
49
|
aggregate_to_cohorts,
|
|
35
50
|
balance_panel,
|
|
@@ -51,6 +66,7 @@ from diff_diff.results import (
|
|
|
51
66
|
from diff_diff.staggered import (
|
|
52
67
|
CallawaySantAnna,
|
|
53
68
|
CallawaySantAnnaResults,
|
|
69
|
+
CSBootstrapResults,
|
|
54
70
|
GroupTimeEffect,
|
|
55
71
|
)
|
|
56
72
|
from diff_diff.utils import (
|
|
@@ -61,13 +77,15 @@ from diff_diff.utils import (
|
|
|
61
77
|
wild_bootstrap_se,
|
|
62
78
|
)
|
|
63
79
|
from diff_diff.visualization import (
|
|
80
|
+
plot_bacon,
|
|
64
81
|
plot_event_study,
|
|
65
82
|
plot_group_effects,
|
|
66
83
|
plot_honest_event_study,
|
|
84
|
+
plot_power_curve,
|
|
67
85
|
plot_sensitivity,
|
|
68
86
|
)
|
|
69
87
|
|
|
70
|
-
__version__ = "0.
|
|
88
|
+
__version__ = "1.0.0"
|
|
71
89
|
__all__ = [
|
|
72
90
|
# Estimators
|
|
73
91
|
"DifferenceInDifferences",
|
|
@@ -75,12 +93,19 @@ __all__ = [
|
|
|
75
93
|
"MultiPeriodDiD",
|
|
76
94
|
"SyntheticDiD",
|
|
77
95
|
"CallawaySantAnna",
|
|
96
|
+
# Bacon Decomposition
|
|
97
|
+
"BaconDecomposition",
|
|
98
|
+
"BaconDecompositionResults",
|
|
99
|
+
"Comparison2x2",
|
|
100
|
+
"bacon_decompose",
|
|
101
|
+
"plot_bacon",
|
|
78
102
|
# Results
|
|
79
103
|
"DiDResults",
|
|
80
104
|
"MultiPeriodDiDResults",
|
|
81
105
|
"SyntheticDiDResults",
|
|
82
106
|
"PeriodEffect",
|
|
83
107
|
"CallawaySantAnnaResults",
|
|
108
|
+
"CSBootstrapResults",
|
|
84
109
|
"GroupTimeEffect",
|
|
85
110
|
# Visualization
|
|
86
111
|
"plot_event_study",
|
|
@@ -122,4 +147,13 @@ __all__ = [
|
|
|
122
147
|
"DeltaSDRM",
|
|
123
148
|
"compute_honest_did",
|
|
124
149
|
"sensitivity_plot",
|
|
150
|
+
# Power analysis
|
|
151
|
+
"PowerAnalysis",
|
|
152
|
+
"PowerResults",
|
|
153
|
+
"SimulationPowerResults",
|
|
154
|
+
"compute_mde",
|
|
155
|
+
"compute_power",
|
|
156
|
+
"compute_sample_size",
|
|
157
|
+
"simulate_power",
|
|
158
|
+
"plot_power_curve",
|
|
125
159
|
]
|