API and Modules

Below is a summary of the main modules:

• Training: Module for training machine learning models.

• Data generator: Handles data generation for training and evaluation.

• Data loaders: Manages datasets for loading and evaluation purposes.

• Models: Contains definitions of the solvers.

• Networks: Defines network architectures used in the models.

• Utils: Utility functions and tools to support model development.

• Preprocess RNA: Preprocessing pipeline for RNA-Seq data.

• Training
• Data generator
• Data loaders
• Models
  • models.base module
  • models.jkonet module
  • models.jkonet_star module
• Networks
  • networks.energies module
  • networks.fixpoint_loop module
  • networks.icnns module
  • networks.optim module
  • networks.utils module
• Utils
  • utils.density module
  • utils.features module
  • utils.functions module
  • utils.ot module
  • utils.plotting module
  • utils.sde_simulator module
• Preprocessing RNA