Struct CateFit 

pub struct CateFit { /* private fields */ }

A fitted CATE model.

Implementations

impl CateFit

pub fn method(&self) -> &str

"S-learner", "T-learner", …

pub fn effect(&self, x: ArrayView2<'_, f64>) -> Result<Array1<f64>>

Estimated effects τ̂(x) for each row of x.

pub fn average_effect(&self, x: ArrayView2<'_, f64>) -> Result<f64>

Average of τ̂(x) over the rows of x.

pub fn pseudo_outcomes(&self) -> Option<ArrayView1<'_, f64>>

DR-learner only: the cross-fitted AIPW pseudo-outcomes φᵢ.

pub fn blp(&self, basis: ArrayView2<'_, f64>) -> Result<Blp>

DR-learner only: best linear predictor of the CATE on basis with HC0 inference. A basis of ones gives the ATE with its DML standard error.

pub fn gate(&self, groups: ArrayView2<'_, f64>) -> Result<Blp>

DR-learner only: group average treatment effects for 0/1 group dummies.

Trait Implementations

impl Debug for CateFit

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.