rbartpackages.dbarts.RunSamples

class rbartpackages.dbarts.RunSamples

Type of the return value of dbarts.run.

Unlike the bart fit attributes, the observations are on the first axis, and multiple chains add a trailing nchain axis.

k: NotRequired[Float64[ndarray, 'ndpost'] | Float64[ndarray, 'ndpost nchain']]

End-node-prior k draws (only with a k hyperprior, the binary default).

sigma: Float64[ndarray, 'ndpost'] | Float64[ndarray, 'ndpost nchain']

Error-SD draws; fixed at 1 for binary responses.

train: Float64[ndarray, 'n ndpost'] | Float64[ndarray, 'n ndpost nchain']

Training-point posterior function draws.

test: Float64[ndarray, 'm ndpost'] | Float64[ndarray, 'm ndpost nchain'] | None

Test-point posterior function draws; None without test data.

varcount: Int32[ndarray, 'p ndpost'] | Int32[ndarray, 'p ndpost nchain']

Per-draw count of splits on each variable, summed over trees.