EpochChainManager#

class liesel.goose.chain.EpochChainManager(apply_thinning=False)[source]#

Bases: Generic[TPyTree]

An EpochChainManager is a container for multiple epoch chains.

The chains can be concatenated over multiple epochs. Thinning defined in epochs can be switched on or of with the constructor flag

Methods

`advance_epoch`(epoch)	Creates and appends a `ListEpochChain` for the given `epoch`.
`append`(chunk)	Applies thinning and appends a chunk to the chain.
`combine`(epoch_numbers)	Combines the given epochs and returns all chunks combined into one pytree.
`combine_all`()	Combines the all epochs and returns all chunks combined into one pytree.
`combine_filtered`(predicate)	Combines the all epochs for which the predicate evaluates to `True`.
`get_current_chain`()	Returns the current chain.
`get_current_epoch`()	Returns the current poch.
`get_epochs`()	Returns a list of all epochs.
`get_specific_chain`(epoch_number)	Returns the chain for the given epoch number.

Attributes

Returns the current epoch.

EpochChainManager