econml.inference.LinearModelFinalInferenceDiscrete
- class econml.inference.LinearModelFinalInferenceDiscrete[source]
Bases:
econml.inference._inference.GenericModelFinalInferenceDiscrete
Inference method for estimators with categorical treatments, where a linear in X model is used for the CATE associated with each treatment. Implements the coef__interval and intercept__interval based on the corresponding methods of the underlying model_final estimator.
- __init__()
Methods
__init__
()ate_inference
([X, T0, T1])ate_interval
([X, T0, T1, alpha])coef__inference
(T)coef__interval
(T, *[, alpha])const_marginal_ate_inference
([X])const_marginal_ate_interval
([X, alpha])const_marginal_effect_inference
(X)const_marginal_effect_interval
(X, *[, alpha])effect_inference
(X, *, T0, T1)effect_interval
(X, *, T0, T1[, alpha])fit
(estimator, *args, **kwargs)Fits the inference model.
intercept__inference
(T)intercept__interval
(T, *[, alpha])marginal_ate_inference
(T[, X])marginal_ate_interval
(T[, X, alpha])prefit
(estimator, *args, **kwargs)Performs any necessary logic before the estimator's fit has been called.
- fit(estimator, *args, **kwargs)
Fits the inference model.
This is called after the estimator’s fit.
- prefit(estimator, *args, **kwargs)
Performs any necessary logic before the estimator’s fit has been called.