Skip to content

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.