nexuml_library.layers.head.regression_head

nexuml_library.layers.head.regression_head

Regression head for latent space.

LatentRegressionHead

Bases: LayerDefinition

Regression head operating on latent vectors.