Skip to content

nexuml_library.layers.loss.reconstruction_loss

nexuml_library.layers.loss.reconstruction_loss

Reconstruction loss layer.

ReconstructionLoss

Bases: LayerDefinition

Computes MSE reconstruction loss between original and reconstructed features.

keys_in should be [original_key, reconstructed_key]. keys_out should be [loss_key].