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].