tmle3
tmle3 copied to clipboard
Categorical outcome support
Does tmle3Â support categorical outcomes? If not, can this be intercepted and and produce an error before fitting SL?
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.