blackjax.adaptation.base

blackjax.adaptation.base#

Classes#

AdaptationResults

AdaptationInfo

Functions#

return_all_adapt_info(state, info, adaptation_state)

Return fully populated AdaptationInfo. Used for adaptation_info_fn

get_filter_adapt_info_fn([state_keys, info_keys, ...])

Generate a function to filter what is saved in AdaptationInfo. Used

Module Contents#

class AdaptationResults[source]#
state: blackjax.types.ArrayTree[source]#
parameters: dict[source]#
class AdaptationInfo[source]#
state: NamedTuple[source]#
info: NamedTuple[source]#
adaptation_state: NamedTuple[source]#
return_all_adapt_info(state, info, adaptation_state)[source]#

Return fully populated AdaptationInfo. Used for adaptation_info_fn parameters of the adaptation algorithms.

get_filter_adapt_info_fn(state_keys: Set[str] = set(), info_keys: Set[str] = set(), adapt_state_keys: Set[str] = set())[source]#

Generate a function to filter what is saved in AdaptationInfo. Used for adptation_info_fn parameters of the adaptation algorithms. adaptation_info_fn=get_filter_adapt_info_fn() saves no auxiliary information