API Reference
CaML Core
from caml import *
Core functionality of CaML for estimating causal effects with cross-sectional data.
| AutoCATE | The AutoCATE class is a high-level API facilitating an AutoML framework for CATE estimation, built on top of the EconML library. |
| InteractiveLinearRegression | InteractiveLinearRegression is an interactive linear regression estimator with explicit treatment interaction terms, enabling precision improvements & heterogeneous treatment discovery. |
Synthetic Data Generation
from caml.extensions.synthetic_data import *
Synthetic data generation utilities for CaML.
| SyntheticDataGenerator | Generate highly flexible synthetic data for use in causal inference and CaML testing. |
| make_partially_linear_dataset_simple | Simulate data generating process from a partially linear model with a simple 1 or 2 dimensional CATE function. |
| make_partially_linear_dataset_constant | Simulate a data generating process from a partially linear model with a constant treatment effect (ATE only). |
| make_fully_heterogeneous_dataset | Simulate data generating process from an interactive regression model with fully heterogenous treatment effects. |
Plots
from caml.extensions.plots import *
Plotting utilities for CaML.
| cate_histogram_plot | Plots a histogram the estimated CATEs. |
| cate_line_plot | Plots a line plot of the ordered estimated CATEs as a rolling mean with optional confidence intervals. |
| cate_true_vs_estimated_plot | Plots a scatter plot of the estimated CATEs against the true CATEs. |
Developer Tools
Developer tools and utilities for CaML.
| decorators | Decorator utilities for CaML. |
| interfaces | Interface utilities for CaML. |
| utils | Various Utilities for CaML. |
| logging | Logging utilities for CaML. |