|
22 | 22 | >>> sp.outreg2(result, filename="results.xlsx") |
23 | 23 | """ |
24 | 24 |
|
25 | | -__version__ = "0.9.17" |
| 25 | +__version__ = "1.0.0" |
26 | 26 | __author__ = "Biaoyue Wang" |
27 | 27 | __email__ = "brycew6m@stanford.edu" |
28 | 28 |
|
|
58 | 58 | rdhte, rdbwhte, rdhte_lincom, |
59 | 59 | rd_forest, rd_boost, rd_lasso, rd_cate_summary, |
60 | 60 | rd_extrapolate, rd_multi_extrapolate, rd_external_validity, |
| 61 | + rd_interference, RDInterferenceResult, |
| 62 | + rd_multi_score, MultiScoreRDResult, |
| 63 | + rd_distribution, DistRDResult, |
| 64 | + rd_bayes_hte, BayesRDHTEResult, |
| 65 | + rd_distributional_design, DDDResult, |
61 | 66 | ) |
62 | 67 | from .synth import ( |
63 | 68 | synth, SyntheticControl, synthplot, sdid, augsynth, |
|
158 | 163 | from .metalearners import auto_cate_tuned |
159 | 164 | from .bayes import ( |
160 | 165 | bayes_did, bayes_rd, bayes_iv, bayes_fuzzy_rd, bayes_hte_iv, |
161 | | - bayes_mte, |
| 166 | + bayes_mte, bayes_dml, BayesianDMLResult, |
162 | 167 | BayesianCausalResult, BayesianDIDResult, BayesianHTEIVResult, |
163 | 168 | BayesianIVResult, BayesianMTEResult, |
164 | 169 | policy_weight_ate, policy_weight_subsidy, |
|
185 | 190 | conformal_ite_multidp, MultiDPConformalResult, |
186 | 191 | conformal_debiased_ml, DebiasedConformalResult, |
187 | 192 | conformal_fair_ite, FairConformalResult, |
| 193 | + conformal_continuous, conformal_interference, |
| 194 | + ContinuousConformalResult, InterferenceConformalResult, |
188 | 195 | ) |
189 | 196 | from .bcf import bcf, BayesianCausalForest, bcf_longitudinal, BCFLongResult |
190 | 197 | from .bunching import bunching, BunchingEstimator, notch, NotchResult |
191 | 198 | from .matrix_completion import mc_panel, MCPanel |
192 | 199 | from .dose_response import dose_response, DoseResponse, vcnet, scigan, VCNetResult |
193 | 200 | from .bounds import lee_bounds, manski_bounds, BoundsResult, horowitz_manski, iv_bounds, oster_delta, selection_bounds, breakdown_frontier, balke_pearl, BalkePearlResult, ml_bounds, MLBoundsResult |
194 | | -from .interference import spillover, SpilloverEstimator, network_exposure, NetworkExposureResult, peer_effects, PeerEffectsResult |
| 201 | +from .interference import spillover, SpilloverEstimator, network_exposure, NetworkExposureResult, peer_effects, PeerEffectsResult, network_hte, inward_outward_spillover, NetworkHTEResult, InwardOutwardResult |
195 | 202 | from .dtr import g_estimation, GEstimation, q_learning, QLearningResult, a_learning, ALearningResult, snmm, SNMMResult |
196 | 203 | from .multi_treatment import multi_treatment, MultiTreatment |
197 | | -from .robustness import spec_curve, SpecCurveResult, robustness_report, RobustnessResult, subgroup_analysis, SubgroupResult |
| 204 | +from .robustness import spec_curve, SpecCurveResult, robustness_report, RobustnessResult, subgroup_analysis, SubgroupResult, copula_sensitivity, survival_sensitivity, calibrate_confounding_strength, FrontierSensitivityResult |
198 | 205 | from .survey import svydesign, SurveyDesign, svymean, svytotal, svyglm, rake, linear_calibration |
199 | 206 | from .dag import ( |
200 | 207 | dag, DAG, dag_example, dag_examples, dag_example_positions, dag_simulate, |
|
209 | 216 | # DR-Calib, Surrogate≡PCI) === |
210 | 217 | from .bridge import bridge, BridgeResult |
211 | 218 |
|
| 219 | +# === LLM × Causal (DAG / E-value / sensitivity priors) === |
| 220 | +from . import causal_llm |
| 221 | +from .causal_llm import ( |
| 222 | + llm_dag_propose, LLMDAGProposal, |
| 223 | + llm_unobserved_confounders, UnobservedConfounderProposal, |
| 224 | + llm_sensitivity_priors, SensitivityPriorProposal, |
| 225 | +) |
| 226 | + |
| 227 | +# === Causal RL (Causal-DQN, benchmarks, offline-safe) === |
| 228 | +from . import causal_rl |
| 229 | +from .causal_rl import ( |
| 230 | + causal_dqn, CausalDQNResult, |
| 231 | + causal_rl_benchmark, BanditBenchmarkResult, |
| 232 | + offline_safe_policy, OfflineSafeResult, |
| 233 | +) |
| 234 | + |
212 | 235 | # === Long-term effects via surrogate indices === |
213 | 236 | from . import surrogate |
214 | 237 | from .surrogate import ( |
|
390 | 413 | SteigerResult, MRPressoResult, RadialResult, |
391 | 414 | mr_mode, mr_f_statistic, mr_funnel_plot, mr_scatter_plot, |
392 | 415 | ModeBasedResult, FStatisticResult, |
| 416 | + mr_multivariable, mr_mediation, mr_bma, |
| 417 | + MVMRResult, MediationMRResult, MRBMAResult, |
393 | 418 | ) |
394 | 419 | # Expose recommend_estimator at top level too |
395 | 420 | from .dag import recommend_estimator as dag_recommend_estimator |
|
1110 | 1135 | "unified_sensitivity", "SensitivityDashboard", |
1111 | 1136 | # v0.9.17 additions (DAG UX) |
1112 | 1137 | "dag_recommend_estimator", |
| 1138 | + # v1.0 — bridging theorems |
| 1139 | + "bridge", "BridgeResult", |
| 1140 | + # v1.0 — DiD frontiers (scaffolded) |
| 1141 | + "did_bcf", "cohort_anchored_event_study", |
| 1142 | + "design_robust_event_study", "did_misclassified", |
| 1143 | + # v1.0 — conformal frontiers |
| 1144 | + "conformal_debiased_ml", "DebiasedConformalResult", |
| 1145 | + "conformal_density_ite", "ConformalDensityResult", |
| 1146 | + "conformal_fair_ite", "FairConformalResult", |
| 1147 | + "conformal_ite_multidp", "MultiDPConformalResult", |
| 1148 | + # v1.0 — proximal frontiers |
| 1149 | + "fortified_pci", "bidirectional_pci", "pci_mtp", |
| 1150 | + "select_pci_proxies", "ProxyScoreResult", |
| 1151 | + # v1.0 — QTE / RD frontiers |
| 1152 | + "beyond_average_late", "BeyondAverageResult", |
| 1153 | + "qte_hd_panel", "HDPanelQTEResult", |
| 1154 | + "rd_distribution", "DistRDResult", |
| 1155 | + "rd_interference", "RDInterferenceResult", |
| 1156 | + "rd_multi_score", "MultiScoreRDResult", |
| 1157 | + # v1.0 — time-series causal discovery |
| 1158 | + "pcmci", "PCMCIResult", "lpcmci", "LPCMCIResult", |
| 1159 | + "dynotears", "DYNOTEARSResult", "partial_corr_pvalue", |
| 1160 | + # v1.0 — LTMLE survival + BCF longitudinal |
| 1161 | + "ltmle_survival", "LTMLESurvivalResult", |
| 1162 | + # v1.0 — sequential SDID |
| 1163 | + "sequential_sdid", "SequentialSDIDResult", |
| 1164 | + # v1.0 — ML bounds |
| 1165 | + "ml_bounds", |
| 1166 | + # v1.0 — TARGET Statement 2025 |
| 1167 | + "target_trial_checklist", |
| 1168 | + # v1.0 — frontier sensitivity |
| 1169 | + "copula_sensitivity", "survival_sensitivity", |
| 1170 | + "calibrate_confounding_strength", "FrontierSensitivityResult", |
1113 | 1171 | ] |
1114 | 1172 |
|
1115 | 1173 |
|
|
0 commit comments