JakeColtman / bartpy

Bayesian Additive Regression Trees For Python
https://jakecoltman.github.io/bartpy/
MIT License
219 stars 44 forks source link

Binary Outcomes and Random Intercepts #57

Open jlevy44 opened 3 years ago

jlevy44 commented 3 years ago

Are there plans to add dichotomous outcomes and random intercepts to this model? Another question is how one would incorporate the random intercepts to conditional means for prediction of the dichotomous outcome? Thanks!

https://github.com/vdorie/dbarts/blob/4ed4eafc772e95d788d5a135d9f4e4728b9516ec/R/rbart.R#L5