scglue.models.scglue.IndSCGLUETrainer

class scglue.models.scglue.IndSCGLUETrainer(net, lam_data=None, lam_kl=None, lam_graph=None, lam_align=None, lam_sup=None, normalize_u=None, domain_weight=None, optim=None, lr=None, **kwargs)[source]

Bases: scglue.models.scglue.SCGLUETrainer

Trainer for IndSCGLUE

Methods

compute_losses

Compute loss functions

train_step

A single training step

Attributes

logger