Index

_

__init__() (scDRP.data.CombinedDataset method) (scDRP.data.LabelDataset method) (scDRP.data.LatentDataset method) (scDRP.loss.HSICLoss method) (scDRP.loss.NOCCOLoss method) (scDRP.loss.UnnormalizedHSCICLoss method) (scDRP.loss.ZINBLoss method) (scDRP.model.Decoder method) (scDRP.model.Encoder method) (scDRP.model.HardConcreteGate method) (scDRP.model.MSEDecoder method) (scDRP.model.NBDecoder method) (scDRP.model.PerturbNet method) (scDRP.module.DosageModel method) (scDRP.module.Perturb method) (scDRP.train.EarlyStopping method) _center_gram() (scDRP.loss.NOCCOLoss method) (scDRP.loss.UnnormalizedHSCICLoss method) _compute_normalized_cov() (scDRP.loss.NOCCOLoss method)

_compute_sigma() (scDRP.loss.HSICLoss method) (scDRP.loss.NOCCOLoss method) (scDRP.loss.UnnormalizedHSCICLoss method) _fit_dose_to_dist() (scDRP.module.Perturb method) _fit_gpr() (scDRP.module.Perturb method) _fit_interpolator() (scDRP.module.Perturb method) _gaussian_kernel() (scDRP.loss.HSICLoss method) (scDRP.loss.NOCCOLoss method) (scDRP.loss.UnnormalizedHSCICLoss method) _generate_from_latent() (scDRP.module.Perturb method) _generate_latent() (scDRP.module.Perturb method) _median_heuristic() (scDRP.loss.HSICLoss method) (scDRP.loss.NOCCOLoss method) (scDRP.loss.UnnormalizedHSCICLoss method) _representative_latent() (scDRP.module.Perturb method) _sample_from_parameter() (scDRP.module.Perturb method) _scotts_rule() (scDRP.loss.HSICLoss method) (scDRP.loss.NOCCOLoss method) (scDRP.loss.UnnormalizedHSCICLoss method)

A

add_effect() (in module scDRP.utils)

C

CombinedDataset (class in scDRP.data) condition_quantile() (in module scDRP.utils)

conditional_OT_latent() (in module scDRP.utils) counterfactual_samples() (scDRP.module.Perturb method) create_dataset() (scDRP.module.Perturb method)

D

Decoder (class in scDRP.model)

dosage_extrapolate() (scDRP.module.Perturb method) DosageModel (class in scDRP.module)

E

EarlyStopping (class in scDRP.train)

effect_estimate() (scDRP.module.Perturb method) Encoder (class in scDRP.model)

F

forward() (scDRP.loss.HSICLoss method) (scDRP.loss.NOCCOLoss method) (scDRP.loss.UnnormalizedHSCICLoss method) (scDRP.loss.ZINBLoss method) (scDRP.model.Decoder method) (scDRP.model.Encoder method) (scDRP.model.HardConcreteGate method) (scDRP.model.MSEDecoder method) (scDRP.model.NBDecoder method) (scDRP.model.PerturbNet method)

G

get_adata() (scDRP.module.Perturb method) get_counterfactual_adata() (scDRP.module.Perturb method)

get_counterfactual_latent() (scDRP.module.Perturb method) get_latent() (scDRP.module.Perturb method) get_parameter_from_latent() (scDRP.module.Perturb method)

H

HardConcreteGate (class in scDRP.model)

HSICLoss (class in scDRP.loss)

I

inference() (scDRP.module.Perturb method)

inference_model() (in module scDRP.train)

J

joint_effect_estimate() (scDRP.module.Perturb method)

K

klLoss() (in module scDRP.loss)

klLoss_prior() (in module scDRP.loss)

L

LabelDataset (class in scDRP.data) latent_adaption() (scDRP.module.Perturb method)

LatentDataset (class in scDRP.data) load() (scDRP.module.Perturb method)

M

module scDRP scDRP.data scDRP.loss scDRP.model scDRP.module scDRP.train scDRP.utils

MSEDecoder (class in scDRP.model)

N

NBDecoder (class in scDRP.model)

NOCCOLoss (class in scDRP.loss)

P

Perturb (class in scDRP.module)

PerturbNet (class in scDRP.model) predict() (scDRP.module.DosageModel method)

R

regularization_loss() (scDRP.model.HardConcreteGate method)

reparameterize() (scDRP.model.Encoder method) (scDRP.model.PerturbNet method)

S

sample_gate() (scDRP.model.HardConcreteGate method) sample_posterior() (scDRP.module.Perturb method) sample_sequencing_depth() (scDRP.model.PerturbNet method) save() (scDRP.module.Perturb method) save_checkpoint() (scDRP.train.EarlyStopping method) scDRP module scDRP.data module scDRP.loss module

scDRP.model module scDRP.module module scDRP.train module scDRP.utils module setup() (scDRP.module.Perturb method)

T

to_dense_array() (in module scDRP.utils)

train() (scDRP.module.Perturb method) train_model() (in module scDRP.train)

U

UnnormalizedHSCICLoss (class in scDRP.loss)

V

validate_model() (in module scDRP.train)

Z

ZINBLoss (class in scDRP.loss)