scvi.external.scviva.NicheLossOutput

scvi.external.scviva.NicheLossOutput#

class scvi.external.scviva.NicheLossOutput(loss, reconstruction_loss=None, kl_local=None, kl_global=None, classification_loss=None, logits=None, true_labels=None, extra_metrics=<factory>, n_obs_minibatch=None, reconstruction_loss_sum=None, kl_local_sum=None, kl_global_sum=None, composition_loss=None, niche_loss=None)[source]#

Bases: LossOutput

Modify loss output to record niche losses.

Attributes table#

composition_loss

niche_loss

loss

extra_metrics

Methods table#

Attributes#

NicheLossOutput.composition_loss: LossRecord | None = None#
NicheLossOutput.niche_loss: LossRecord | None = None#
NicheLossOutput.loss: LossRecord#
NicheLossOutput.extra_metrics: dict[str, Tensor] | None#

Methods#

Contents