nexuml_library.layers.head.classification_head¶
nexuml_library.layers.head.classification_head
¶
Classification head for latent space.
LatentClassificationHead
¶
Bases: LayerDefinition
Classification head operating on latent vectors.
Produces logits for multiclass or multilabel classification.