larry._callbacks._interpolation_task¶
Module Contents¶
Classes¶
Attributes¶
- larry._callbacks._interpolation_task.__module_name__ = _interpolation_task.py¶
- larry._callbacks._interpolation_task.__doc__ = Callback for passing the test set to measure interpolation of a withheld timepoint.¶
- larry._callbacks._interpolation_task.__author__¶
- larry._callbacks._interpolation_task.__email__¶
- class larry._callbacks._interpolation_task.InterpolationTask(adata: anndata.AnnData, potential: bool = False, time_key: str = 'Time point', lineage_key: str = 'clone_idx', use_key: str = 'X_pca', seed: int = 617, silent: bool = True, device=AutoDevice())¶
Bases:
pytorch_lightning.Callback- configure_eval_data()¶
- forward_without_grad(DiffEq)¶
Forward integrate over the model without gradients.
- forward_with_grad(DiffEq)¶
Forward integrate over the model retaining gradients.
- on_train_epoch_end(trainer, DiffEq)¶
Call