flax_identity_logit_model

probly.transformation.calibration.flax_identity_logit_model() FlaxIdentityLogitModel[source]

Create a flax pass-through model for direct logit calibration.