API Reference
CaML Core
from caml import *
| CamlCATE | The CamlCATE class represents an opinionated framework of Causal Machine Learning techniques for estimating highly accurate conditional average treatment effects (CATEs). |
Synthetic Data Generation
from caml.extensions.synthetic_data import *
| CamlSyntheticDataGenerator | 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 *
| 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
| generics.generate_random_string | Function to generate a random string of ascii lowercase letters and digits of length N. |
| generics.cls_typechecked | Class decorator to typecheck all methods of a class. |
| logging.setup_logging | Set up logging configuration. |