diff-diff 0.2.0__py3-none-any.whl → 0.4.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 +65 -3
- diff_diff/estimators.py +539 -28
- diff_diff/prep.py +1338 -0
- diff_diff/results.py +234 -26
- diff_diff/staggered.py +982 -0
- diff_diff/utils.py +357 -18
- diff_diff/visualization.py +471 -0
- diff_diff-0.4.0.dist-info/METADATA +1394 -0
- diff_diff-0.4.0.dist-info/RECORD +11 -0
- diff_diff-0.2.0.dist-info/METADATA +0 -547
- diff_diff-0.2.0.dist-info/RECORD +0 -8
- {diff_diff-0.2.0.dist-info → diff_diff-0.4.0.dist-info}/WHEEL +0 -0
- {diff_diff-0.2.0.dist-info → diff_diff-0.4.0.dist-info}/top_level.txt +0 -0
diff_diff/__init__.py
CHANGED
|
@@ -5,14 +5,76 @@ 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.estimators import
|
|
9
|
-
|
|
8
|
+
from diff_diff.estimators import (
|
|
9
|
+
DifferenceInDifferences,
|
|
10
|
+
TwoWayFixedEffects,
|
|
11
|
+
MultiPeriodDiD,
|
|
12
|
+
SyntheticDiD,
|
|
13
|
+
)
|
|
14
|
+
from diff_diff.staggered import (
|
|
15
|
+
CallawaySantAnna,
|
|
16
|
+
CallawaySantAnnaResults,
|
|
17
|
+
GroupTimeEffect,
|
|
18
|
+
)
|
|
19
|
+
from diff_diff.results import (
|
|
20
|
+
DiDResults,
|
|
21
|
+
MultiPeriodDiDResults,
|
|
22
|
+
PeriodEffect,
|
|
23
|
+
SyntheticDiDResults,
|
|
24
|
+
)
|
|
25
|
+
from diff_diff.visualization import (
|
|
26
|
+
plot_event_study,
|
|
27
|
+
plot_group_effects,
|
|
28
|
+
)
|
|
29
|
+
from diff_diff.prep import (
|
|
30
|
+
make_treatment_indicator,
|
|
31
|
+
make_post_indicator,
|
|
32
|
+
wide_to_long,
|
|
33
|
+
balance_panel,
|
|
34
|
+
validate_did_data,
|
|
35
|
+
summarize_did_data,
|
|
36
|
+
generate_did_data,
|
|
37
|
+
create_event_time,
|
|
38
|
+
aggregate_to_cohorts,
|
|
39
|
+
rank_control_units,
|
|
40
|
+
)
|
|
41
|
+
from diff_diff.utils import (
|
|
42
|
+
check_parallel_trends,
|
|
43
|
+
check_parallel_trends_robust,
|
|
44
|
+
equivalence_test_trends,
|
|
45
|
+
)
|
|
10
46
|
|
|
11
|
-
__version__ = "0.
|
|
47
|
+
__version__ = "0.4.0"
|
|
12
48
|
__all__ = [
|
|
49
|
+
# Estimators
|
|
13
50
|
"DifferenceInDifferences",
|
|
51
|
+
"TwoWayFixedEffects",
|
|
14
52
|
"MultiPeriodDiD",
|
|
53
|
+
"SyntheticDiD",
|
|
54
|
+
"CallawaySantAnna",
|
|
55
|
+
# Results
|
|
15
56
|
"DiDResults",
|
|
16
57
|
"MultiPeriodDiDResults",
|
|
58
|
+
"SyntheticDiDResults",
|
|
17
59
|
"PeriodEffect",
|
|
60
|
+
"CallawaySantAnnaResults",
|
|
61
|
+
"GroupTimeEffect",
|
|
62
|
+
# Visualization
|
|
63
|
+
"plot_event_study",
|
|
64
|
+
"plot_group_effects",
|
|
65
|
+
# Parallel trends testing
|
|
66
|
+
"check_parallel_trends",
|
|
67
|
+
"check_parallel_trends_robust",
|
|
68
|
+
"equivalence_test_trends",
|
|
69
|
+
# Data preparation utilities
|
|
70
|
+
"make_treatment_indicator",
|
|
71
|
+
"make_post_indicator",
|
|
72
|
+
"wide_to_long",
|
|
73
|
+
"balance_panel",
|
|
74
|
+
"validate_did_data",
|
|
75
|
+
"summarize_did_data",
|
|
76
|
+
"generate_did_data",
|
|
77
|
+
"create_event_time",
|
|
78
|
+
"aggregate_to_cohorts",
|
|
79
|
+
"rank_control_units",
|
|
18
80
|
]
|