Individual Treatment Effects

[PrannoyLFC]

Hi @kbattocchi ,

I used the following code to calculate the ATE for my panel data (around ~$18).

dml = DynamicDML(model_y=outcome_model,
model_t=treatment_model,
cv=2, discrete_treatment=True)

dml.fit(Y=Y, T=T, X=None, W=X, groups=groups)

Is it possible to get individual treatment effects too? Ie how much causal impact the treatment would gave in the outcome variable for each row of my data?

[ellpri]

@PrannoyLFC I think, in order to get the effect for each row, you need to use dml.effect(x_test)Reference