cellflow.data.ConditionData

class cellflow.data.ConditionData(condition_data, max_combination_length, perturbation_idx_to_covariates, perturbation_idx_to_id, null_value, data_manager)

Data container containing condition embeddings.

Parameters:

• condition_data (dict[str, ndarray]) – Dictionary with embeddings for conditions.

• max_combination_length (int) – Maximum number of covariates in a combination.

• null_value (Any) – Token to use for masking null_value.

• data_manager (Any) – Data manager used to generate the data.

• perturbation_idx_to_covariates (dict[int, tuple[str, ...]])

• perturbation_idx_to_id (dict[int, Any])

Methods

Attributes

n_controls	Returns the number of control covariate values.
n_perturbation_covariates	Returns the number of perturbation covariates.
n_perturbations	Returns the number of perturbation covariate combinations.
condition_data
max_combination_length
perturbation_idx_to_covariates
perturbation_idx_to_id
null_value
data_manager