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.
Back to top