Skip to content

supaernova.configs.steps.posterior.tf

source module supaernova.configs.steps.posterior.tf

Classes

Functions

source validate_loss(loss: ConfigInputObject[LossObject])

Raises

  • ValueError

source get_loss(loss_fn: Callable[[tf.Tensor, tf.Tensor], tf.Tensor])type[ks.losses.Loss]

source class TFPosteriorModelConfig(**data: Any)

Bases : PosteriorModelConfig

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.

Attributes

  • model_extra : dict[str, Any] | None Get extra fields set during validation.

  • model_fields_set : set[str] Returns the set of fields that have been explicitly set on this model instance.

source property TFPosteriorModelConfig.loss_cls: type[ks.losses.Loss] | None