boax.core.objectives.negative_log_likelihood

boax.core.objectives.negative_log_likelihood(logpdf_fn)

The negative log likelihood objective function.

Example

>>> objective = negative_log_likelihood(logpdf_fn)
>>> nll = objective(prediction, targets)

Parameters:

logpdf_fn (Callable[[TypeVar(T), Union[Array, ndarray, bool, number]], Union[Array, ndarray, bool, number]]) – The log probability mass function.

Return type:

Objective[TypeVar(T)]

Returns:

The negative log likelihood Objective function.