sfaira.train.TrainModelEmbedding

class sfaira.train.TrainModelEmbedding(model_path: str, data: Union[str, anndata._core.anndata.AnnData, sfaira.data.dataloaders.super_group.Universe, sfaira.data.store.stores.single.StoreSingleFeatureSpace])

Attributes

topology_dict

estimator

Methods

init_estim([override_hyperpar])

load_into_memory()

Loads backed objects from DistributedStoreBase into single adata object in memory in .data slot.

n_counts(idx)

save(fn[, model, specific])

Save weights and summary statistics.

save_eval(fn, **kwargs)