CausalML: Python Package for Causal Machine Learning
Reviewed by Pithpith:HS5M64NSopen to challenge →
read the original abstract
CausalML is a Python implementation of algorithms related to causal inference and machine learning. Algorithms combining causal inference and machine learning have been a trending topic in recent years. This package tries to bridge the gap between theoretical work on methodology and practical applications by making a collection of methods in this field available in Python. This paper introduces the key concepts, scope, and use cases of this package.
This paper has not been read by Pith yet.
Forward citations
Cited by 9 Pith papers
-
ConfoundingSHAP: Quantifying confounding strength in causal inference
ConfoundingSHAP defines a custom Shapley game to attribute confounding strength to individual covariates and uses TabPFN to estimate it scalably without exhaustive refitting.
-
Differential Subgroup Discovery: Characterizing Where Two Populations Differ, and Why
Differential subgroups identify specific feature combinations where population differences in outcomes are most extreme, found via a new optimization objective and the DiffSub method.
-
Real vs. Semi-Simulated: Rethinking Evaluation for Treatment Effect Estimation
Counterfactual metrics on semi-simulated benchmarks fail to identify the treatment effect estimators preferred by observable metrics on real datasets, with simple meta-learners outperforming specialized causal models.
-
Real vs. Semi-Simulated: Rethinking Evaluation for Treatment Effect Estimation
Large-scale study finds that counterfactual metrics on semi-simulated data do not select the same estimators as observable metrics on real data, and benchmark rankings fail to transfer.
-
CounterBench: Evaluating and Improving Counterfactual Reasoning in Large Language Models
Introduces CounterBench benchmark and CoIn iterative reasoning method showing LLMs perform near random on formal counterfactual tasks but improve substantially with guided backtracking.
-
Improving Bias Correction Standards by Quantifying its Effects on Treatment Outcomes
Introduces the A2A metric to evaluate propensity score matching methods by constructing artificial tasks with known outcomes, reducing ATE errors up to 50% when combined with SMD.
-
Personalizing Marketplace Policies with Competing Objectives and Constrained Experiments: Evidence from a Job Marketplace
An integrated framework using ensemble hybrid ranking models and monotonicity-based extrapolation enables personalized policy deployment in a constrained two-sided job marketplace, achieving significant target metric ...
-
Spark Policy Toolkit: Semantic Contracts and Scalable Execution for Policy Learning in Spark
Spark Policy Toolkit supplies semantic contracts plus mapInPandas/mapInArrow inference and executor-side split search so policy learning remains correct and fast on Spark clusters up to tens of millions of rows.
-
Closing the Loop: A Software Framework for AI to Support Business Decision Making
A software framework integrates heterogeneous causal inference, policy learning, mediation, forecasts, variance reduction, and anytime-valid inference into one AI-orchestratable interface for business experimentation.
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.