bartz.mcmcloop.StatsReport

class bartz.mcmcloop.StatsReport(grow_prop, move_acc, mean_leaves, peff, n_samples, num_chains, max_leaves, p)

Forest diagnostics produced by StatsAccumulator.report for one report.

grow_prop: Float32[Array, '']

Fraction of trees proposed for a grow move.

move_acc: Float32[Array, '']

Fraction of trees on which a grow or prune move was accepted.

mean_leaves: Float32[Array, '']

Mean number of leaves per tree.

peff: Float32[Array, ''] | None

Effective number of predictors, or None when variable selection is off.

n_samples: Int32[Array, ''] | None

Number of iterations averaged over, or None when not averaging.

num_chains: int | None

Number of chains averaged over, or None when single-chain.

max_leaves: int

Maximum possible number of leaves per tree.

p: int | None

Number of predictors, or None when variable selection is off.