cellpin.models.CellPin.pretrain_model

CellPin.pretrain_model(dataset, custom_callbacks=None, train_size=0.8, pretrain_epochs=50, **trainer_kwargs)

Stage-1 pretraining (full-gene view only, ELBO).

Args:

dataset: Training dataset. custom_callbacks: Extra PyTorch-Lightning callbacks. train_size: Fraction of data used for training. pretrain_epochs: Default max epochs (overridden by

trainer_kwargs['max_epochs'] if present).

**trainer_kwargs: Forwarded to CellPinTrainer.

Returns:

:

Fitted CellPinTrainer.