transformation

Transformations for models.

Submodules

batchensemble(base[, num_members, ...])

Create a BatchEnsemble predictor from a base predictor based on [WTB20].

bayesian(base[, use_base_weights, ...])

Create a Bayesian predictor from a base predictor based on [BCKW15].

bayesian_ensemble(base[, use_base_weights, ...])

Create an ensemble of Bayesian-layer transformed predictors based on [BCKW15].

calibration

Calibration transformations for logit predictors.

cast(base)

Return a predictor unchanged while optionally registering its predictor type.

class_bias_ensemble(base, num_members[, ...])

Create an ensemble with class-specific final-layer bias initialization based on [LohrHMHullermeier25].

conformal

Conformal transformations for regression and classification.

conformal_credal_set

Conformalized Credal Set Prediction implementation.

dirichlet_clipped_exp_one_activation(base)

Append clipped exp plus one so logits become Dirichlet concentrations.

dirichlet_exp_activation(base)

Append an exp activation so logits become Dirichlet concentrations.

dirichlet_softplus_activation(base)

Append Softplus plus one so logits become Dirichlet concentrations.

dropconnect(base[, p, rng_collection, rngs])

Create a DropConnect predictor from a base predictor based on [MYM+21].

dropout(base[, p, rng_collection, rngs, ...])

Create a Dropout predictor from a base predictor based on [GG16].

ensemble(base, num_members[, reset_params])

Create an ensemble predictor from a base predictor based on [LPB17].

interval_classifier(base[, use_base_weights])

Create an interval classifier from a base classifier based on [WCM+24].

natural_posterior_network

Module for natural posterior network implementations.

normal_inverse_gamma_head(base)

Replace the final linear layer with a Normal-Inverse-Gamma head.

posterior_network(encoder, latent_dim, ...)

Create a Posterior Network predictor from an encoder based on [CZugnerGunnemann20].

subensemble(base, num_heads[, head, ...])

Create a subensemble predictor from a base model or a base model and head model.

transformation

Utilities for the definition of methods.