tmle3 icon indicating copy to clipboard operation
tmle3 copied to clipboard

Categorical outcome support

Open rachaelvp opened this issue 3 years ago • 1 comments

Does tmle3 support categorical outcomes? If not, can this be intercepted and and produce an error before fitting SL?

rachaelvp avatar Jan 15 '22 00:01 rachaelvp

No parameters have been implemented that support categorical outcomes, but it’d be trivial to implement ATE or TSM generalizations for them. There’s nothing stopping you from fitting a likelihood with categorical nodes. The likelihood model doesn’t really have a concept of an outcome. That’s all up to the parameters.

I’m not sure where you could add a data type check as you’re suggesting.

jeremyrcoyle avatar Jan 15 '22 01:01 jeremyrcoyle