ModelMixin.log_prob_fn()#

ModelMixin.log_prob_fn(model_state)[source]#

Returns the log-probability function with the position as the only argument.

Return type:

Callable[[NewType(Position, dict[str, Any])], float]