Do auxiliary distributions are used to predict outcome?

WeijiaZhang24 / TEDVAE

Code for TEDVAE, a VAE-based treatment effect estimation algorithm.

20 stars 9 forks source link

Do auxiliary distributions are used to predict outcome? #1

Closed WeilinChen507 closed 3 years ago

WeilinChen507 commented 3 years ago

description:

Hello, I am using the codes of TEDVAE. In the paper, it points out that the predicted outcomes y are inferred using the auxiliary classifier in the inference model (the auxiliary distributions in the Guide part in codes), but the predicted outcomes y seems to be inferred from the generative model (the outcome from the Model part in codes). Could u tell me which distribution (in Guide or in Model?) are used to inferred y? Thanks very much.

WeijiaZhang24 commented 3 years ago

It should be the generative model.

WeilinChen507 commented 3 years ago

It should be the generative model.

Thanks very much.