No packages match

grf - Generalized Random Forests

Forest-based statistical estimation and inference. GRF provides non-parametric methods for heterogeneous treatment effects estimation (optionally using right-censored outcomes, multiple treatment arms or outcomes, or instrumental variables), as well as least-squares regression, quantile regression, and survival regression, all with support for missing covariates.

Last updated

cpp

7.62 score 27 dependents 2.3k scripts 11k downloads

policytree - Policy Learning via Doubly Robust Empirical Welfare Maximization over Trees

Learn optimal policies via doubly robust empirical welfare maximization over trees. Given doubly robust reward estimates, this package finds a rule-based treatment prescription policy, where the policy takes the form of a shallow decision tree that is globally (or close to) optimal.

Last updated

cpp

3.84 score 4 dependents 119 scripts 4.9k downloads

balnet - Pathwise Estimation of Covariate Balancing Propensity Scores

Provides pathwise estimation of regularized logistic propensity score models using covariate balancing loss functions rather than maximum likelihood. Regularization paths are fit via the 'adelie' elastic-net solver with a 'glmnet'-like interface, yielding balancing weights that target covariate balance for the ATE and ATT. Under lasso penalization, lambda bounds the maximum covariate imbalance, so the regularization path traces a sequence of decreasing imbalance tolerances. For details, see Sverdrup & Hastie (2026) <doi:10.48550/arXiv.2602.18577>.

Last updated

cppopenmp

2.48 score 6 scripts 342 downloads

maq - Multi-Armed Qini

Policy evaluation using generalized Qini curves: Evaluate data-driven treatment targeting rules for one or more treatment arms over different budget constraints in experimental or observational settings under unconfoundedness.

Last updated

cpp

1.46 score 29 scripts 161 downloads