causalfe: Causal Forests with Fixed Effects in Python
By: Harry Aytug
The causalfe package provides a Python implementation of Causal Forests with Fixed Effects (CFFE) for estimating heterogeneous treatment effects in panel data settings. Standard causal forest methods struggle with panel data because unit and time fixed effects induce spurious heterogeneity in treatment effect estimates. The CFFE approach addresses this by performing node-level residualization during tree construction, removing fixed effects within each candidate split rather than globally. This paper describes the methodology, documents the software interface, and demonstrates the package through simulation studies that validate the estimator's performance under various data generating processes.
Similar Papers
Causal-Policy Forest for End-to-End Policy Learning
Econometrics
Helps doctors pick best treatment for each person.
Multi-Study Causal Forest (MCF): A flexible framework for data borrowing in the presence of varying treatment effect heterogeneity
Methodology
Helps doctors pick best medicine for each person.
Branching Fixed Effects: A Proposal for Communicating Uncertainty
Econometrics
Helps check if economic studies are correct.