setu.MixedNLETrainingResult

class setu.MixedNLETrainingResult(mixed_nle, losses, val_losses, best_epoch)

Bases: object

Result of MixedNLE training.

mixed_nle

Trained MixedNLE instance with fitted transforms.

losses

Training loss per epoch.

val_losses

Validation loss per epoch.

best_epoch

Epoch with lowest validation loss.

Parameters:

• mixed_nle (MixedNLE)

• losses (Array)

• val_losses (Array)

• best_epoch (int)

property estimator: MixedNLE

The trained estimator (alias for .mixed_nle, uniform across result types).

validate(val_data, key, *, checks=None, strict=False, config=None)

Run validation checks on the trained estimator.

Parameters:

• val_data (ValidationDataset) – Validation dataset (separate from training data).

• key (Array) – JAX random key.

• checks (list[str] | None) – List of checks to run (default: all).

• strict (bool) – If True, raise on failure.

• config (ValidationConfig | None) – Threshold configuration.

Return type:

ValidationResult