scmg.model.causal_prediction module

class scmg.model.causal_prediction.CausalGenePredictor(adata_pert, gene_stds)

Bases: object

Parameters:

adata_pert (AnnData) – Annotated data object with the perturbation shifts.

calc_causal_scores(mean_exp_shifts)

Parameters:

exp_shifts (np.ndarray) – The mean expression shifts for the cell state transition.

Returns:

A DataFrame with the causal scores for each perturbation.

Return type:

pd.DataFrame