User API

• Model
  • cellflow.model.CellFlow
• Solvers
  • cellflow.solvers.OTFlowMatching
  • cellflow.solvers.GENOT
• Networks
  • cellflow.networks.ConditionalVelocityField
  • cellflow.networks.GENOTConditionalVelocityField
  • cellflow.networks.ConditionEncoder
  • cellflow.networks.SelfAttention
  • cellflow.networks.SeedAttentionPooling
  • cellflow.networks.TokenAttentionPooling
  • cellflow.networks.MLPBlock
  • cellflow.networks.SelfAttentionBlock
  • cellflow.networks.BaseModule
  • cellflow.networks.FilmBlock
  • cellflow.networks.ResNetBlock
  • cellflow.networks.SelfAttentionBlock
• Datasets
  • cellflow.datasets.ineurons
  • cellflow.datasets.pbmc_cytokines
• Metrics
  • cellflow.metrics.compute_metrics
  • cellflow.metrics.compute_mean_metrics
  • cellflow.metrics.compute_scalar_mmd
  • cellflow.metrics.compute_r_squared
  • cellflow.metrics.compute_sinkhorn_div
  • cellflow.metrics.compute_e_distance
  • cellflow.metrics.maximum_mean_discrepancy
• Utils
  • cellflow.utils.match_linear
• Training
  • cellflow.training.BaseCallback
  • cellflow.training.CallbackRunner
  • cellflow.training.ComputationCallback
  • cellflow.training.LoggingCallback
  • cellflow.training.Metrics
  • cellflow.training.PCADecodedMetrics
  • cellflow.training.VAEDecodedMetrics
  • cellflow.training.WandbLogger
  • cellflow.training.CellFlowTrainer
• Plotting
  • cellflow.plotting.plot_condition_embedding
• Preprocessing
  • cellflow.preprocessing.centered_pca
  • cellflow.preprocessing.project_pca
  • cellflow.preprocessing.reconstruct_pca
  • cellflow.preprocessing.annotate_compounds
  • cellflow.preprocessing.encode_onehot
  • cellflow.preprocessing.get_molecular_fingerprints
  • cellflow.preprocessing.compute_wknn
  • cellflow.preprocessing.transfer_labels
• External
  • cellflow.external.CFJaxSCVI