Struct CausalKernel 

pub struct CausalKernel<State = Untrained> { /* private fields */ }

Causal Kernel for Treatment Effect Estimation

Implements kernel-based methods for estimating causal effects from observational data, including propensity score weighting and doubly robust estimation.

Mathematical Background

For treatment T, outcome Y, and covariates X:

• ATE = E[Y(1) - Y(0)] = E[Y|do(T=1)] - E[Y|do(T=0)]
• Uses kernel-based representation: τ(x) = E[Y|T=1,X=x] - E[Y|T=0,X=x]

Examples

use sklears_kernel_approximation::causal_kernels::{CausalKernel, CausalKernelConfig};
use scirs2_core::ndarray::array;
use sklears_core::traits::{Fit, Transform};

let config = CausalKernelConfig::default();
let causal = CausalKernel::new(config);

// X: covariates, T: treatment, Y: outcome
let data = array![[1.0, 0.0, 2.0], [2.0, 1.0, 5.0]];
let fitted = causal.fit(&data, &()).unwrap();
let features = fitted.transform(&data).unwrap();

Implementations

impl CausalKernel<Untrained>

pub fn new(config: CausalKernelConfig) -> Self

Create a new causal kernel

pub fn with_components(n_components: usize) -> Self

Create with default configuration

pub fn method(self, method: CausalMethod) -> Self

Set causal method

pub fn treatment_bandwidth(self, gamma: Float) -> Self

Set treatment bandwidth

impl CausalKernel<Trained>

pub fn propensity_scores(&self) -> &Array1<Float>

Get estimated propensity scores

pub fn treatment_effects(&self) -> &HashMap<String, Float>

Get estimated treatment effects

pub fn ate(&self) -> Float

Get average treatment effect

Trait Implementations

impl<State: Clone> Clone for CausalKernel<State>

fn clone(&self) -> CausalKernel<State>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<State: Debug> Debug for CausalKernel<State>

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

Formats the value using the given formatter.

impl Estimator for CausalKernel<Untrained>

type Config = CausalKernelConfig

Configuration type for the estimator

type Error = SklearsError

Error type for the estimator

type Float = f64

The numeric type used by this estimator

fn config(&self) -> &Self::Config

Get estimator configuration

fn validate_config(&self) -> Result<(), SklearsError>

Validate estimator configuration with detailed error context

fn check_compatibility( &self, n_samples: usize, n_features: usize, ) -> Result<(), SklearsError>

Check if estimator is compatible with given data dimensions

fn metadata(&self) -> EstimatorMetadata

Get estimator metadata

impl Fit<ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>, ()> for CausalKernel<Untrained>

type Fitted = CausalKernel<Trained>

The fitted model type

fn fit(self, x: &Array2<Float>, _y: &()) -> Result<Self::Fitted>

Fit the model to the provided data with validation

fn fit_with_validation( self, x: &X, y: &Y, _x_val: Option<&X>, _y_val: Option<&Y>, ) -> Result<(Self::Fitted, FitMetrics), SklearsError>

where Self: Sized,

Fit with custom validation and early stopping

impl Transform<ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>> for CausalKernel<Trained>

fn transform(&self, x: &Array2<Float>) -> Result<Array2<Float>>

Transform the input data