Index

A

AirQuality (class in gents.dataset.modules) ALLOW_CONDITION (gents.model.base.BaseModel attribute) (gents.model.diffeq.LatentODE attribute) (gents.model.diffeq.LatentSDE attribute) (gents.model.diffeq.LS4 attribute) (gents.model.diffeq.SDEGAN attribute) (gents.model.diffusion.CSDI attribute) (gents.model.diffusion.DiffusionTS attribute) (gents.model.diffusion.FIDE attribute) (gents.model.diffusion.FourierDiffusion attribute) (gents.model.diffusion.ImagenTime attribute) (gents.model.diffusion.TMDM attribute) (gents.model.diffusion.VanillaDDPM attribute) (gents.model.flow.FourierFlow attribute) (gents.model.flow.VanillaMAF attribute) (gents.model.gan.COSCIGAN attribute) (gents.model.gan.GTGAN attribute) (gents.model.gan.PSAGAN attribute) (gents.model.gan.RCGAN attribute) (gents.model.gan.TimeGAN attribute) (gents.model.gan.VanillaGAN attribute) (gents.model.vae.KoVAE attribute) (gents.model.vae.TimeVAE attribute) (gents.model.vae.TimeVQVAE attribute) (gents.model.vae.VanillaVAE attribute)

B

BaseDataModule (class in gents.dataset.base)

BaseModel (class in gents.model.base)

C

configure_optimizers() (gents.model.diffeq.LatentODE method) (gents.model.diffeq.LatentSDE method) (gents.model.diffeq.LS4 method) (gents.model.diffeq.SDEGAN method) (gents.model.diffusion.CSDI method) (gents.model.diffusion.DiffusionTS method) (gents.model.diffusion.FIDE method) (gents.model.diffusion.FourierDiffusion method) (gents.model.diffusion.ImagenTime method) (gents.model.diffusion.TMDM method) (gents.model.diffusion.VanillaDDPM method) (gents.model.flow.FourierFlow method) (gents.model.flow.VanillaMAF method) (gents.model.gan.COSCIGAN method) (gents.model.gan.GTGAN method) (gents.model.gan.PSAGAN method) (gents.model.gan.RCGAN method) (gents.model.gan.TimeGAN method) (gents.model.gan.VanillaGAN method) (gents.model.vae.KoVAE method) (gents.model.vae.TimeVQVAE method) (gents.model.vae.VanillaVAE method)

context_fid() (in module gents.evaluation.model_based) COSCIGAN (class in gents.model.gan) crps() (in module gents.evaluation.model_free) CSDI (class in gents.model.diffusion) csv_dir (gents.dataset.base.WebDownloadDataModule attribute) (gents.dataset.modules.Electricity attribute) (gents.dataset.modules.Exchange attribute) (gents.dataset.modules.Traffic attribute) (gents.dataset.modules.Weather attribute)

D

D (gents.dataset.base.WebDownloadDataModule attribute) (gents.dataset.modules.AirQuality attribute) (gents.dataset.modules.ECG attribute) (gents.dataset.modules.Electricity attribute) (gents.dataset.modules.Energy attribute) (gents.dataset.modules.ETTh1 attribute) (gents.dataset.modules.Exchange attribute) (gents.dataset.modules.MuJoCo attribute) (gents.dataset.modules.Physionet attribute) (gents.dataset.modules.Stocks attribute) (gents.dataset.modules.Traffic attribute) (gents.dataset.modules.Weather attribute)

data_source (gents.dataset.base.WebDownloadDataModule attribute) (gents.dataset.modules.Electricity attribute) (gents.dataset.modules.Energy attribute) (gents.dataset.modules.ETTh1 attribute) (gents.dataset.modules.Exchange attribute) (gents.dataset.modules.Stocks attribute) (gents.dataset.modules.Traffic attribute) (gents.dataset.modules.Weather attribute) degrade() (gents.model.diffusion.VanillaDDPM method) DiffusionTS (class in gents.model.diffusion) directional_distance() (gents.evaluation.model_free.WassersteinDistances method) discriminative_score() (in module gents.evaluation.model_based)

E

ECG (class in gents.dataset.modules) Electricity (class in gents.dataset.modules) Energy (class in gents.dataset.modules) ETTh1 (class in gents.dataset.modules)

ETTh2 (class in gents.dataset.modules) ETTm1 (class in gents.dataset.modules) ETTm2 (class in gents.dataset.modules) Exchange (class in gents.dataset.modules)

F

feat_name (gents.dataset.modules.AirQuality attribute) feature_distance() (gents.evaluation.model_free.WassersteinDistances method) FIDE (class in gents.model.diffusion) forward() (gents.model.diffusion.DiffusionTS method) (gents.model.diffusion.FourierDiffusion method) (gents.model.diffusion.FourierDiffusionLSTM method) (gents.model.diffusion.FourierDiffusionMLP method) (gents.model.diffusion.ImagenTime method) (gents.model.flow.FourierFlow method) (gents.model.vae.KoVAE method)

FourierDiffusion (class in gents.model.diffusion) FourierDiffusionLSTM (class in gents.model.diffusion) FourierDiffusionMLP (class in gents.model.diffusion) FourierFlow (class in gents.model.flow)

G

gents module gents.common module gents.dataset.base module gents.dataset.modules module gents.evaluation.model_based module gents.evaluation.model_free module gents.evaluation.visualization module gents.model.base module

gents.model.diffeq module gents.model.diffusion module gents.model.flow module gents.model.gan module gents.model.vae module get_data() (gents.dataset.base.WebDownloadDataModule method) get_marginal_directions() (gents.evaluation.model_free.WassersteinDistances method) get_random_directions() (gents.evaluation.model_free.WassersteinDistances method) GTGAN (class in gents.model.gan)

I

ImagenTime (class in gents.model.diffusion) imputation_visual() (in module gents.evaluation.visualization) index_col (gents.dataset.base.WebDownloadDataModule attribute) (gents.dataset.modules.Electricity attribute) (gents.dataset.modules.Energy attribute) (gents.dataset.modules.ETTh1 attribute) (gents.dataset.modules.Exchange attribute) (gents.dataset.modules.Stocks attribute) (gents.dataset.modules.Traffic attribute) (gents.dataset.modules.Weather attribute)

K

KoVAE (class in gents.model.vae)

L

L (gents.dataset.modules.ECG attribute) (gents.dataset.modules.Physionet attribute) LatentODE (class in gents.model.diffeq)

LatentSDE (class in gents.model.diffeq) loss_fn (gents.model.diffusion.DiffusionTS property) LS4 (class in gents.model.diffeq)

M

marginal_distances() (gents.evaluation.model_free.WassersteinDistances method) module gents gents.common gents.dataset.base gents.dataset.modules gents.evaluation.model_based gents.evaluation.model_free gents.evaluation.visualization gents.model.base gents.model.diffeq gents.model.diffusion gents.model.flow gents.model.gan gents.model.vae

mse() (in module gents.evaluation.model_free) MuJoCo (class in gents.dataset.modules)

N

n_classes (gents.dataset.modules.ECG attribute) (gents.dataset.modules.Spiral2D attribute)

O

on_before_zero_grad() (gents.model.diffusion.DiffusionTS method) (gents.model.diffusion.ImagenTime method) on_fit_end() (gents.model.diffeq.SDEGAN method) (gents.model.diffusion.DiffusionTS method)

on_fit_start() (gents.model.diffusion.FIDE method) (gents.model.diffusion.ImagenTime method) (gents.model.flow.FourierFlow method) on_train_batch_start() (gents.model.diffeq.LatentODE method) on_train_epoch_start() (gents.model.gan.PSAGAN method)

P

Physionet (class in gents.dataset.modules) predict_dataloader() (gents.dataset.base.BaseDataModule method) predict_visual() (in module gents.evaluation.visualization)

predictive_score() (in module gents.evaluation.model_based) prepare_data() (gents.dataset.base.BaseDataModule method) PSAGAN (class in gents.model.gan)

R

random_direction() (gents.evaluation.model_free.WassersteinDistances method)

RCGAN (class in gents.model.gan)

S

sample() (gents.model.base.BaseModel method) sample_loop() (gents.model.diffusion.FIDE method) SDEGAN (class in gents.model.diffeq) setup() (gents.dataset.base.BaseDataModule method)

SineND (class in gents.dataset.modules) sliced_distances() (gents.evaluation.model_free.WassersteinDistances method) Spiral2D (class in gents.dataset.modules) Stocks (class in gents.dataset.modules)

T

test_dataloader() (gents.dataset.base.BaseDataModule method) TimeGAN (class in gents.model.gan) TimeVAE (class in gents.model.vae) TimeVQVAE (class in gents.model.vae) TMDM (class in gents.model.diffusion) Traffic (class in gents.dataset.modules) train_dataloader() (gents.dataset.base.BaseDataModule method) training_step() (gents.model.diffeq.LatentODE method) (gents.model.diffeq.LatentSDE method) (gents.model.diffeq.LS4 method) (gents.model.diffeq.SDEGAN method) (gents.model.diffusion.CSDI method) (gents.model.diffusion.DiffusionTS method) (gents.model.diffusion.FIDE method) (gents.model.diffusion.FourierDiffusion method) (gents.model.diffusion.ImagenTime method) (gents.model.diffusion.TMDM method) (gents.model.diffusion.VanillaDDPM method) (gents.model.flow.FourierFlow method) (gents.model.flow.VanillaMAF method) (gents.model.gan.COSCIGAN method) (gents.model.gan.GTGAN method) (gents.model.gan.PSAGAN method) (gents.model.gan.RCGAN method) (gents.model.gan.TimeGAN method) (gents.model.gan.VanillaGAN method) (gents.model.vae.KoVAE method) (gents.model.vae.TimeVQVAE method) (gents.model.vae.VanillaVAE method)

TSDataset (class in gents.dataset.base) tsne_visual() (in module gents.evaluation.visualization)

U

url (gents.dataset.base.WebDownloadDataModule attribute) (gents.dataset.modules.Physionet attribute) urls (gents.dataset.modules.AirQuality attribute) (gents.dataset.modules.ECG attribute) (gents.dataset.modules.Electricity attribute) (gents.dataset.modules.Energy attribute) (gents.dataset.modules.ETTh1 attribute) (gents.dataset.modules.Exchange attribute) (gents.dataset.modules.Stocks attribute) (gents.dataset.modules.Traffic attribute) (gents.dataset.modules.Weather attribute)

V

val_dataloader() (gents.dataset.base.BaseDataModule method) validation_step() (gents.model.diffeq.LatentODE method) (gents.model.diffeq.LatentSDE method) (gents.model.diffeq.LS4 method) (gents.model.diffeq.SDEGAN method) (gents.model.diffusion.CSDI method) (gents.model.diffusion.DiffusionTS method) (gents.model.diffusion.FIDE method) (gents.model.diffusion.FourierDiffusion method) (gents.model.diffusion.ImagenTime method) (gents.model.diffusion.TMDM method) (gents.model.diffusion.VanillaDDPM method) (gents.model.flow.FourierFlow method) (gents.model.flow.VanillaMAF method) (gents.model.gan.COSCIGAN method) (gents.model.gan.GTGAN method) (gents.model.gan.PSAGAN method) (gents.model.gan.RCGAN method) (gents.model.gan.TimeGAN method) (gents.model.gan.VanillaGAN method) (gents.model.vae.KoVAE method) (gents.model.vae.TimeVQVAE method) (gents.model.vae.VanillaVAE method)

VanillaDDPM (class in gents.model.diffusion) VanillaGAN (class in gents.model.gan) VanillaMAF (class in gents.model.flow) VanillaVAE (class in gents.model.vae)

W

WassersteinDistances (class in gents.evaluation.model_free)

Weather (class in gents.dataset.modules) WebDownloadDataModule (class in gents.dataset.base)