Module model 

DML models with linear scores: Plr, Irm, Pliv, and the LinearScoreModel trait for adding your own.

Structs

Did

Difference-in-differences with panel data (two periods): y must be the outcome change Y₁ − Y₀. ml_g learns E[ΔY | D = d, X] per group, ml_m the propensity P(D = 1 | X) (observational score only).

DidCs

Difference-in-differences with repeated cross-sections: the data need a 0/1 time indicator (DmlData::with_time_indicator). ml_g learns E[Y | D = d, T = t, X] separately in each of the four group × period cells, ml_m the propensity P(D = 1 | X) (observational score only).

Iivm

IIVM estimator of the LATE: ml_g learns E[Y | Z, X] (fitted separately on the Z = 0 and Z = 1 rows), ml_m the instrument propensity P(Z = 1 | X), and ml_r the treatment probability P(D = 1 | Z, X) (again per instrument arm).

Irm

IRM estimator: ml_g learns E[Y | D = d, X] (fitted separately on the treated and control rows of each training fold), ml_m learns the propensity score P(D = 1 | X).

Pliv

PLIV estimator with the partialling-out score ψ = (Y − ℓ(X) − θ(D − r(X)))·(Z − m(X)).

Plr

PLR estimator: ml_l learns E[Y | X], ml_m learns E[D | X], and the optional ml_g learns g(X) for the IV-type score.

ScoreEstimate

Score elements plus the cross-fitted nuisance predictions behind them.

ScoreInput

The data a model sees when estimating the effect of one treatment.

Enums

DidScore

Assignment mechanism assumed by Did / DidCs.

IrmScore

Target parameter of Irm.

PlrScore

Orthogonal score used by Plr.

Traits

LinearScoreModel

A DML model whose orthogonal score is linear in θ.