cellflow.data.DataManager.get_prediction_data

DataManager.get_prediction_data(adata, sample_rep, covariate_data, rep_dict=None, condition_id_key=None)

Get predictions for control cells.

Extracts source distributions from 'adata' and simulates cells perturbed with covariates defined in 'covariate_data'.

Parameters:

• adata (AnnData) – An AnnData object to extract control cells from.

• sample_rep (str) – Key in obsm where the sample representation of the control is stored or 'X' to use X.

• covariate_data (DataFrame) – A DataFrame with columns defining the covariates as in cellflow.model.CellFlow.prepare_data() and stored in cellflow.model.CellFlow.data_manager.

• rep_dict (dict[str, Any] | None) – Dictionary with representations of the covariates. If not provided, uns is used.

• condition_id_key (str | None) – Key in DataFrame that defines the condition names.

Return type:

Any

Returns:

: Training data for the model.