normal_inverse_gamma_head¶ probly.transformation.normal_inverse_gamma_head(base: T) → T[source]¶ Replace the final linear layer with a Normal-Inverse-Gamma head. Parameters: base – The base model to be transformed. Returns: The transformed predictor.