bartz.stochtree

stochtree-compatible interface to bartz.

BARTModel()	BART model with a stochtree-compatible interface, powered by bartz.
GeneralParams(*[, standardize, sigma2_init, ...])	Mirror of stochtree's general_params dict, with the keys bartz handles.
OutcomeModel([outcome, link])	Outcome model specification, matching stochtree.OutcomeModel.
MeanForestParams(*[, num_trees, alpha, ...])	Mirror of stochtree's mean_forest_params dict, restricted to the keys bartz handles.
NotSampledError	Raised when calling a method that requires BARTModel.sample to have been called.