nexuml_library.layers.loss.multilabel_map_metrics¶
Multi-label mAP metric layer.
Bases: LayerDefinition
Pipeline layer computing multi-label mean Average Precision.
Consumes logits from keys_in[0] and multi-hot labels from y[label_key]. Applies sigmoid to logits before updating the wrapped torchmetrics MultilabelAveragePrecision metric.