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 givenepoch
.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.
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
.Returns the current chain.
Returns the current poch.
Returns a list of all epochs.
get_specific_chain
(epoch_number)Returns the chain for the given epoch number.
Attributes
Returns the current epoch.