Deep-Learning-for-Causal-Inference icon indicating copy to clipboard operation
Deep-Learning-for-Causal-Inference copied to clipboard

Published 20 hours ago • verified publisher icon kochbj

Tutorial for multi treatments

Open CoteDave opened this issue 2 years ago • 1 comments

Hi,

Very usefull and great tutorials! Any plans to add a multi treatments tutorial with tarnet or dragonnet ?

Thanks !

CoteDave avatar Oct 14 '22 23:10 CoteDave

Hi,

Thank you for your interest! I had not planned to add additional tutorials at this time; so busy with other things. But hopefully this is helpful:

  1. Should be fairly straightforward to adapt TarNet for multiple treatments. Simply add more heads and make sure the loss only feeds the head that is being used.

  2. For adaptation of Dragonnet, check out these comments by Victor Veitch: https://github.com/claudiashi57/dragonnet/issues/4

Feel free to respond, if you have specific questions and I'll do my best to answer.

All the best, Bernie

kochbj avatar Oct 17 '22 16:10 kochbj