jesterTOV.inference.config.schema.BlackJAXNSAWConfig#

class BlackJAXNSAWConfig(**data)[source]#

Bases: BaseSamplerConfig

Configuration for BlackJAX Nested Sampling with Acceptance Walk.

Variables:
  • type (Literal["blackjax-ns-aw"]) – Sampler type identifier

  • n_live (int) – Number of live points (default: 1000)

  • n_delete_frac (float) – Fraction of live points to delete per iteration (default: 0.5)

  • n_target (int) – Target number of accepted MCMC steps (default: 60)

  • max_mcmc (int) – Maximum MCMC steps per iteration (default: 5000)

  • max_proposals (int) – Maximum proposal attempts per MCMC step (default: 1000)

  • termination_dlogz (float) – Evidence convergence criterion (default: 0.1)

  • output_dir (str) – Directory to save results

  • n_eos_samples (int) – Number of EOS samples to generate after inference (default: 10000)

__init__(**data)#

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Methods

__init__(**data)

Create a new model by parsing and validating input data from keyword arguments.

construct([_fields_set])

copy(*[, include, exclude, update, deep])

Returns a copy of the model.

dict(*[, include, exclude, by_alias, ...])

from_orm(obj)

json(*[, include, exclude, by_alias, ...])

model_construct([_fields_set])

Creates a new instance of the Model class with validated data.

model_copy(*[, update, deep])

!!! abstract "Usage Documentation"

model_dump(*[, mode, include, exclude, ...])

!!! abstract "Usage Documentation"

model_dump_json(*[, indent, ensure_ascii, ...])

!!! abstract "Usage Documentation"

model_json_schema([by_alias, ref_template, ...])

Generates a JSON schema for a model class.

model_parametrized_name(params)

Compute the class name for parametrizations of generic classes.

model_post_init(context, /)

Override this method to perform additional initialization after __init__ and model_construct.

model_rebuild(*[, force, raise_errors, ...])

Try to rebuild the pydantic-core schema for the model.

model_validate(obj, *[, strict, extra, ...])

Validate a pydantic model instance.

model_validate_json(json_data, *[, strict, ...])

!!! abstract "Usage Documentation"

model_validate_strings(obj, *[, strict, ...])

Validate the given object with string data against the Pydantic model.

parse_file(path, *[, content_type, ...])

parse_obj(obj)

parse_raw(b, *[, content_type, encoding, ...])

schema([by_alias, ref_template])

schema_json(*[, by_alias, ref_template])

update_forward_refs(**localns)

validate(value)

validate_base_positive(v)

Validate that value is positive.

validate_delete_frac(v)

Validate that deletion fraction is in (0, 1].

validate_positive(v)

Validate that value is positive.

Attributes

model_computed_fields

model_config

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

model_extra

Get extra fields set during validation.

model_fields

model_fields_set

Returns the set of fields that have been explicitly set on this model instance.

type

n_live

n_delete_frac

n_target

max_mcmc

max_proposals

termination_dlogz

output_dir

n_eos_samples

log_prob_batch_size

max_mcmc: int#
max_proposals: int#
model_config: ClassVar[ConfigDict] = {'extra': 'forbid'}#

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

n_delete_frac: float#
n_live: int#
n_target: int#
termination_dlogz: float#
type: Literal['blackjax-ns-aw']#
classmethod validate_delete_frac(v)[source]#

Validate that deletion fraction is in (0, 1].

Return type:

float

classmethod validate_positive(v)[source]#

Validate that value is positive.

Return type:

int