bartz.stochtree.OutcomeModel

class bartz.stochtree.OutcomeModel(outcome='continuous', link=None)

Outcome model specification, matching stochtree.OutcomeModel.

Only ('continuous', 'identity') and ('binary', 'probit') are supported.

outcome: Literal['continuous', 'binary'] = 'continuous'

Outcome family.

link: Literal['identity', 'probit'] | None = None

Link function. If None, defaults to 'identity' for 'continuous' and 'probit' for 'binary'.