scglue.models.scglue.SCGLUEModel.compile
- SCGLUEModel.compile(lam_data=1.0, lam_kl=1.0, lam_graph=0.02, lam_align=0.05, lam_sup=0.02, normalize_u=False, modality_weight=None, lr=2e-3, **kwargs)[source]
Prepare model for training
- Parameters
lam_data (
float
) – Data weightlam_kl (
float
) – KL weightlam_graph (
float
) – Graph weightlam_align (
float
) – Adversarial alignment weightlam_sup (
float
) – Cell type supervision weightnormalize_u (
bool
) – Whether to L2 normalize cell embeddings before decodermodality_weight (
typing.Optional
[typing.Mapping
[str
,float
]]) – Relative modality weight (indexed by modality name)lr (
float
) – Learning rate**kwargs – Additional keyword arguments passed to trainer
- Return type