EpochChainManager#

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

Bases: Generic[liesel.goose.chain.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

current_epoch

Returns the current epoch.