bartz.testing.QuantizedData¶

class bartz.testing.QuantizedData(x, y, max_split)[source]¶

Output of DGP.quantize: data in the format of bartz.mcmcstep.init.

x: UInt[Array, 'p n']¶: Quantized predictors of shape (p, n), with values in [0, max_split[j]] in row j.

y: Float[Array, 'k n'] | Float[Array, 'n']¶: Outcomes, same as DGP.y.

max_split: UInt[Array, 'p']¶: Number of allowed cutpoints per predictor.