multi_ellipsoidal_samplers

jaxns.samplers.multi_ellipsoidal_samplers

Module Contents

class MultiEllipsoidalSampler(depth, expansion_factor, *args, **kwargs)

Bases: jaxns.samplers.bases.BaseAbstractRejectionSampler

Uses a multi-ellipsoidal decomposition of the live points to create a bound around regions to sample from.

Inefficient for high dimensional problems, but can be very efficient for low dimensional problems.

Parameters:

• depth (int) –

• expansion_factor (float) –

property max_num_ellipsoids

num_phantom()

Return type:

int

pre_process(state)

Parameters:

state (jaxns.internals.types.StaticStandardNestedSamplerState) –

Return type:

jaxns.samplers.abc.SamplerState

post_process(sample_collection, sampler_state)

Parameters:

• sample_collection (jaxns.internals.types.StaticStandardSampleCollection) –

• sampler_state (jaxns.samplers.abc.SamplerState) –

Return type:

jaxns.samplers.abc.SamplerState

get_sample(key, log_L_constraint, sampler_state)

Parameters:

• key (jaxns.internals.types.PRNGKey) –

• log_L_constraint (jaxns.internals.types.FloatArray) –

• sampler_state (jaxns.samplers.multi_ellipsoid.multi_ellipsoid_utils.MultEllipsoidState) –

Return type:

Tuple[jaxns.internals.types.Sample, jaxns.internals.types.Sample]