Developer API

CellFlow model

DataManager(adata, sample_rep, control_key)

Data manager for handling perturbation data.

TrainingData(cell_data, ...)

Training data.

PredictionData(cell_data, ...)

Data container to perform prediction.

ValidationData(cell_data, ...[, ...])

Data container for the validation data.

ConditionData(condition_data, ...)

Data container containing condition embeddings.

BaseDataMixin()

Base class for data containers.

TrainSampler(data[, batch_size])

Data sampler for TrainingData.

ValidationSampler(val_data[, seed])

Data sampler for ValidationData.

PredictionSampler(pred_data)

Data sampler for PredictionData.