supaernova.configs.steps.pae.tf

source module supaernova.configs.steps.pae.tf

Classes

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

Functions

• validate_activation

• validate_kernel_regulariser

• validate_scheduler

• validate_optimiser

• validate_loss

• get_loss

source validate_activation(activation: ConfigInputObject[ActivationObject])

source validate_kernel_regulariser(kernel_regulariser: ConfigInputObject[RegulariserObject]) → RegulariserObject

source validate_scheduler(scheduler: ConfigInputObject[SchedulerObject]) → SchedulerObject

source validate_optimiser(optimiser: ConfigInputObject[OptimiserObject])

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 TFPAEModelConfig(**data: Any)

Bases : PAEModelConfig

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 TFPAEModelConfig.activation_fn: ActivationObject

source property TFPAEModelConfig.kernel_regulariser_cls: type[ks.regularizers.Regularizer] | None

source property TFPAEModelConfig.scheduler_cls: type[ks.optimizers.schedules.LearningRateSchedule]

source property TFPAEModelConfig.optimiser_cls: type[ks.optimizers.Optimizer]

source property TFPAEModelConfig.loss_cls: type[ks.losses.Loss]