estimate_policy: (Re)Estimate treatment and response surface models for a...

In jongbinjung/undi: Test for Unjustified Disparate Impact

Description

(Re)Estimate treatment and response surface models for a policy

Usage

estimate_policy(pol, features = NULL, ptreat = pol$ptreat,
  resp_ctl = pol$resp_ctl, resp_trt = pol$resp_trt,
  calibrate = pol$calibrate, save_models = pol$save_models, ...)

Arguments

pol	a policy object
features	(Optional) character vector of features to use; if NULL, pol$features is used
ptreat	(Optional) default value for probability of treatment;
resp_ctl	(Optional)
resp_trt	(Optional) default value for probability of response = 1 given each treatment regime (ctl, trt); useful for cases where outcome under certain treatment regimes is deterministic (e.g., probability of finding illegal weapon if NOT frisked is 0)
calibrate	whether or not to use platt scaling to calibrate predictions
save_models	whether or not fitted models should be returned
...	additional arguments passed to model fitting functions fit1 and fit_ptreat. See policy

Value

object of class policy where the relative columns in $data are updated with new predictions

jongbinjung/undi documentation built on May 8, 2019, 11:56 p.m.