NeuroDiME: A Software Library on Neural-based Estimation of Divergences and Metrics
Contents:
models
NeuroDiME: A Software Library on Neural-based Estimation of Divergences and Metrics
Index
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
I
|
J
|
K
|
L
|
M
|
N
|
P
|
R
|
S
|
T
|
X
|
Z
_
__call__() (GAN_CIFAR10_jax.Discriminator method)
,
[1]
(GAN_CIFAR10_jax.Discriminator_cond method)
,
[1]
(GAN_CIFAR10_jax.Generator method)
,
[1]
(GAN_CIFAR10_jax.Generator_cond method)
,
[1]
A
alpha_Divergence_LT (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
avg_pool2d() (in module GAN_CIFAR10_tf)
,
[1]
(in module GAN_CIFAR10_torch)
,
[1]
B
bounded (model_jax.Discriminator attribute)
bounded_activation() (model_jax.Discriminator method)
,
[1]
(model_tf.Discriminator method)
,
[1]
,
[2]
,
[3]
(model_torch.BoundedActivation method)
,
[1]
BoundedActivation (class in model_torch)
,
[1]
build() (GAN_CIFAR10_tf.Discriminator method)
,
[1]
(GAN_CIFAR10_tf.DiscriminatorBlock method)
,
[1]
(GAN_CIFAR10_tf.Generator method)
,
[1]
(GAN_CIFAR10_tf.GeneratorBlock method)
,
[1]
C
call() (GAN_CIFAR10_tf.Discriminator method)
,
[1]
,
[2]
,
[3]
(GAN_CIFAR10_tf.DiscriminatorBlock method)
,
[1]
,
[2]
,
[3]
(GAN_CIFAR10_tf.Generator method)
,
[1]
,
[2]
,
[3]
(GAN_CIFAR10_tf.GeneratorBlock method)
,
[1]
,
[2]
,
[3]
(model_tf.Discriminator method)
,
[1]
,
[2]
,
[3]
(model_tf.DiscriminatorMNIST method)
,
[1]
,
[2]
,
[3]
(model_tf.Generator method)
,
[1]
,
[2]
,
[3]
compute_output_shape() (GAN_CIFAR10_tf.GeneratorBlock method)
,
[1]
D
DataLoader (class in Divergences_jax)
,
[1]
(class in GAN_jax)
disc_train_step() (GAN_jax.GAN method)
,
[1]
(GAN_tf.GAN method)
,
[1]
(GAN_torch.GAN method)
,
[1]
discriminate() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
Discriminator (class in GAN_CIFAR10_jax)
,
[1]
(class in GAN_CIFAR10_tf)
,
[1]
(class in GAN_CIFAR10_torch)
,
[1]
(class in model_jax)
,
[1]
(class in model_tf)
,
[1]
(class in model_torch)
,
[1]
Discriminator_cond (class in GAN_CIFAR10_jax)
,
[1]
(class in GAN_CIFAR10_torch)
,
[1]
discriminator_loss() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
Discriminator_MNIST (class in GAN_MNIST_torch)
,
[1]
(class in model_torch)
,
[1]
Discriminator_MNIST_cond (class in GAN_MNIST_jax)
,
[1]
(class in GAN_MNIST_torch)
,
[1]
Discriminator_MNIST_cond() (in module GAN_MNIST_tf)
,
[1]
Discriminator_Penalty (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
DiscriminatorBlock (class in GAN_CIFAR10_tf)
,
[1]
(class in GAN_CIFAR10_torch)
,
[1]
DiscriminatorMNIST (class in model_jax)
,
[1]
(class in model_tf)
,
[1]
Divergence (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
Divergences_jax
module
,
[1]
Divergences_tf
module
,
[1]
Divergences_torch
module
,
[1]
E
estimate() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
estimate_loss() (GAN_jax.GAN method)
,
[1]
(GAN_tf.GAN method)
,
[1]
(GAN_torch.GAN method)
,
[1]
eval_var_formula() (Divergences_jax.Divergence method)
,
[1]
(Divergences_jax.f_Divergence method)
,
[1]
(Divergences_jax.IPM method)
,
[1]
(Divergences_jax.KLD_DV method)
,
[1]
(Divergences_jax.Pearson_chi_squared_HCR method)
,
[1]
(Divergences_jax.Renyi_Divergence_CC method)
,
[1]
(Divergences_jax.Renyi_Divergence_CC_rescaled method)
,
[1]
(Divergences_jax.Renyi_Divergence_DV method)
,
[1]
(Divergences_jax.Renyi_Divergence_WCR method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_tf.f_Divergence method)
,
[1]
(Divergences_tf.IPM method)
,
[1]
(Divergences_tf.KLD_DV method)
,
[1]
(Divergences_tf.Pearson_chi_squared_HCR method)
,
[1]
(Divergences_tf.Renyi_Divergence_CC method)
,
[1]
(Divergences_tf.Renyi_Divergence_CC_rescaled method)
,
[1]
(Divergences_tf.Renyi_Divergence_DV method)
,
[1]
(Divergences_tf.Renyi_Divergence_WCR method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
(Divergences_torch.f_Divergence method)
,
[1]
(Divergences_torch.IPM method)
,
[1]
(Divergences_torch.KLD_DV method)
,
[1]
(Divergences_torch.Pearson_chi_squared_HCR method)
,
[1]
(Divergences_torch.Renyi_Divergence_CC method)
,
[1]
(Divergences_torch.Renyi_Divergence_CC_rescaled method)
,
[1]
(Divergences_torch.Renyi_Divergence_DV method)
,
[1]
(Divergences_torch.Renyi_Divergence_WCR method)
,
[1]
eval_var_formula_gen() (Divergences_jax.f_Divergence method)
,
[1]
(Divergences_jax.IPM method)
,
[1]
(Divergences_jax.KLD_DV method)
,
[1]
(Divergences_jax.Pearson_chi_squared_HCR method)
,
[1]
(Divergences_jax.Renyi_Divergence_CC method)
,
[1]
(Divergences_jax.Renyi_Divergence_CC_rescaled method)
,
[1]
(Divergences_jax.Renyi_Divergence_DV method)
,
[1]
(Divergences_jax.Renyi_Divergence_WCR method)
,
[1]
(Divergences_tf.f_Divergence method)
,
[1]
(Divergences_tf.IPM method)
,
[1]
(Divergences_tf.KLD_DV method)
,
[1]
(Divergences_tf.Pearson_chi_squared_HCR method)
,
[1]
(Divergences_tf.Renyi_Divergence_CC method)
,
[1]
(Divergences_tf.Renyi_Divergence_CC_rescaled method)
,
[1]
(Divergences_tf.Renyi_Divergence_DV method)
,
[1]
(Divergences_tf.Renyi_Divergence_WCR method)
,
[1]
(Divergences_torch.f_Divergence method)
,
[1]
(Divergences_torch.IPM method)
,
[1]
(Divergences_torch.KLD_DV method)
,
[1]
(Divergences_torch.Pearson_chi_squared_HCR method)
,
[1]
(Divergences_torch.Renyi_Divergence_CC method)
,
[1]
(Divergences_torch.Renyi_Divergence_CC_rescaled method)
,
[1]
(Divergences_torch.Renyi_Divergence_DV method)
,
[1]
(Divergences_torch.Renyi_Divergence_WCR method)
,
[1]
evaluate() (Divergences_jax.Discriminator_Penalty method)
,
[1]
(Divergences_jax.Gradient_Penalty_1Sided method)
,
[1]
(Divergences_jax.Gradient_Penalty_2Sided method)
,
[1]
(Divergences_tf.Discriminator_Penalty method)
,
[1]
(Divergences_tf.Gradient_Penalty_1Sided method)
,
[1]
(Divergences_tf.Gradient_Penalty_2Sided method)
,
[1]
(Divergences_torch.Discriminator_Penalty method)
,
[1]
(Divergences_torch.Gradient_Penalty_1Sided method)
,
[1]
(Divergences_torch.Gradient_Penalty_2Sided method)
,
[1]
F
f_Divergence (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
f_star() (Divergences_jax.alpha_Divergence_LT method)
,
[1]
(Divergences_jax.f_Divergence method)
,
[1]
(Divergences_jax.Jensen_Shannon_LT method)
,
[1]
(Divergences_jax.KLD_LT method)
,
[1]
(Divergences_jax.Pearson_chi_squared_LT method)
,
[1]
(Divergences_jax.squared_Hellinger_LT method)
,
[1]
(Divergences_tf.alpha_Divergence_LT method)
,
[1]
(Divergences_tf.f_Divergence method)
,
[1]
(Divergences_tf.Jensen_Shannon_LT method)
,
[1]
(Divergences_tf.KLD_LT method)
,
[1]
(Divergences_tf.Pearson_chi_squared_LT method)
,
[1]
(Divergences_tf.squared_Hellinger_LT method)
,
[1]
(Divergences_torch.alpha_Divergence_LT method)
,
[1]
(Divergences_torch.f_Divergence method)
,
[1]
(Divergences_torch.Jensen_Shannon_LT method)
,
[1]
(Divergences_torch.KLD_LT method)
,
[1]
(Divergences_torch.Pearson_chi_squared_LT method)
,
[1]
(Divergences_torch.squared_Hellinger_LT method)
,
[1]
final_layer_activation() (Divergences_jax.f_Divergence method)
,
[1]
(Divergences_jax.Jensen_Shannon_LT method)
,
[1]
(Divergences_jax.Renyi_Divergence_CC method)
,
[1]
(Divergences_jax.Renyi_Divergence_CC_rescaled method)
,
[1]
(Divergences_jax.squared_Hellinger_LT method)
,
[1]
(Divergences_tf.f_Divergence method)
,
[1]
(Divergences_tf.Jensen_Shannon_LT method)
,
[1]
(Divergences_tf.Renyi_Divergence_CC method)
,
[1]
(Divergences_tf.Renyi_Divergence_CC_rescaled method)
,
[1]
(Divergences_tf.squared_Hellinger_LT method)
,
[1]
(Divergences_torch.f_Divergence method)
,
[1]
(Divergences_torch.Jensen_Shannon_LT method)
,
[1]
(Divergences_torch.Renyi_Divergence_CC method)
,
[1]
(Divergences_torch.Renyi_Divergence_CC_rescaled method)
,
[1]
(Divergences_torch.squared_Hellinger_LT method)
,
[1]
forward() (GAN_CIFAR10_torch.Discriminator method)
,
[1]
,
[2]
,
[3]
(GAN_CIFAR10_torch.Discriminator_cond method)
,
[1]
,
[2]
,
[3]
(GAN_CIFAR10_torch.DiscriminatorBlock method)
,
[1]
,
[2]
,
[3]
(GAN_CIFAR10_torch.Generator method)
,
[1]
,
[2]
,
[3]
(GAN_CIFAR10_torch.Generator_cond method)
,
[1]
,
[2]
,
[3]
(GAN_CIFAR10_torch.GeneratorBlock method)
,
[1]
,
[2]
,
[3]
(GAN_MNIST_torch.Discriminator_MNIST method)
,
[1]
,
[2]
,
[3]
(GAN_MNIST_torch.Discriminator_MNIST_cond method)
,
[1]
,
[2]
,
[3]
(GAN_MNIST_torch.Generator_MNIST method)
,
[1]
,
[2]
,
[3]
(GAN_MNIST_torch.Generator_MNIST_cond method)
,
[1]
,
[2]
,
[3]
(model_torch.BoundedActivation method)
,
[1]
,
[2]
,
[3]
(model_torch.Discriminator method)
,
[1]
,
[2]
,
[3]
(model_torch.Discriminator_MNIST method)
,
[1]
,
[2]
,
[3]
(model_torch.Generator method)
,
[1]
,
[2]
,
[3]
from_config() (GAN_CIFAR10_tf.GeneratorBlock class method)
,
[1]
G
GAN (class in GAN_jax)
,
[1]
(class in GAN_tf)
,
[1]
(class in GAN_torch)
,
[1]
GAN_CIFAR10_jax
module
,
[1]
GAN_CIFAR10_tf
module
,
[1]
GAN_CIFAR10_torch
module
,
[1]
GAN_jax
module
,
[1]
GAN_MNIST_jax
module
,
[1]
GAN_MNIST_tf
module
,
[1]
GAN_MNIST_torch
module
,
[1]
GAN_tf
module
,
[1]
GAN_torch
module
,
[1]
gen_train_step() (Divergences_jax.Divergence method)
,
[1]
(GAN_jax.GAN method)
,
[1]
(GAN_tf.GAN method)
,
[1]
(GAN_torch.GAN method)
,
[1]
generate_samples() (GAN_jax.GAN method)
(GAN_tf.GAN method)
(GAN_torch.GAN method)
Generator (class in GAN_CIFAR10_jax)
,
[1]
(class in GAN_CIFAR10_tf)
,
[1]
(class in GAN_CIFAR10_torch)
,
[1]
(class in model_jax)
,
[1]
(class in model_tf)
,
[1]
(class in model_torch)
,
[1]
Generator_cond (class in GAN_CIFAR10_jax)
,
[1]
(class in GAN_CIFAR10_torch)
,
[1]
generator_loss() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
Generator_MNIST (class in GAN_MNIST_torch)
,
[1]
Generator_MNIST_cond (class in GAN_MNIST_jax)
,
[1]
(class in GAN_MNIST_torch)
,
[1]
Generator_MNIST_cond() (in module GAN_MNIST_tf)
,
[1]
GeneratorBlock (class in GAN_CIFAR10_tf)
,
[1]
(class in GAN_CIFAR10_torch)
,
[1]
get_batch_size() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
get_config() (GAN_CIFAR10_tf.GeneratorBlock method)
,
[1]
get_discriminator() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
get_learning_rate() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
get_Lip_constant() (Divergences_jax.Gradient_Penalty_1Sided method)
,
[1]
(Divergences_jax.Gradient_Penalty_2Sided method)
,
[1]
(Divergences_tf.Gradient_Penalty_1Sided method)
,
[1]
(Divergences_tf.Gradient_Penalty_2Sided method)
,
[1]
(Divergences_torch.Gradient_Penalty_1Sided method)
,
[1]
(Divergences_torch.Gradient_Penalty_2Sided method)
,
[1]
get_no_epochs() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
get_order() (Divergences_jax.alpha_Divergence_LT method)
,
[1]
(Divergences_jax.Renyi_Divergence method)
,
[1]
(Divergences_tf.alpha_Divergence_LT method)
,
[1]
(Divergences_tf.Renyi_Divergence method)
,
[1]
(Divergences_torch.alpha_Divergence_LT method)
,
[1]
(Divergences_torch.Renyi_Divergence method)
,
[1]
get_penalty_weight() (Divergences_jax.Discriminator_Penalty method)
,
[1]
(Divergences_tf.Discriminator_Penalty method)
,
[1]
(Divergences_torch.Discriminator_Penalty method)
,
[1]
Gradient_Penalty_1Sided (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
Gradient_Penalty_2Sided (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
I
input_dim (model_jax.Discriminator attribute)
IPM (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
J
Jensen_Shannon_LT (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
K
KLD_DV (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
KLD_LT (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
L
latent_dim (GAN_MNIST_jax.Generator_MNIST_cond attribute)
layers_list (model_jax.Discriminator attribute)
(model_jax.Generator attribute)
M
model_jax
module
,
[1]
model_tf
module
,
[1]
model_torch
module
,
[1]
module
Divergences_jax
,
[1]
Divergences_tf
,
[1]
Divergences_torch
,
[1]
GAN_CIFAR10_jax
,
[1]
GAN_CIFAR10_tf
,
[1]
GAN_CIFAR10_torch
,
[1]
GAN_jax
,
[1]
GAN_MNIST_jax
,
[1]
GAN_MNIST_tf
,
[1]
GAN_MNIST_torch
,
[1]
GAN_tf
,
[1]
GAN_torch
,
[1]
model_jax
,
[1]
model_tf
,
[1]
model_torch
,
[1]
N
name (GAN_CIFAR10_jax.Discriminator attribute)
(GAN_CIFAR10_jax.Discriminator_cond attribute)
(GAN_CIFAR10_jax.Generator attribute)
(GAN_CIFAR10_jax.Generator_cond attribute)
(GAN_MNIST_jax.Discriminator_MNIST_cond attribute)
(GAN_MNIST_jax.Generator_MNIST_cond attribute)
(model_jax.Discriminator attribute)
(model_jax.DiscriminatorMNIST attribute)
(model_jax.Generator attribute)
num_classes (GAN_MNIST_jax.Generator_MNIST_cond attribute)
P
parent (GAN_CIFAR10_jax.Discriminator attribute)
(GAN_CIFAR10_jax.Discriminator_cond attribute)
(GAN_CIFAR10_jax.Generator attribute)
(GAN_CIFAR10_jax.Generator_cond attribute)
(GAN_MNIST_jax.Discriminator_MNIST_cond attribute)
(GAN_MNIST_jax.Generator_MNIST_cond attribute)
(model_jax.Discriminator attribute)
(model_jax.DiscriminatorMNIST attribute)
(model_jax.Generator attribute)
Pearson_chi_squared_HCR (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
Pearson_chi_squared_LT (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
R
Renyi_Divergence (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
Renyi_Divergence_CC (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
Renyi_Divergence_CC_rescaled (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
Renyi_Divergence_DV (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
Renyi_Divergence_WCR (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
S
scope (GAN_CIFAR10_jax.Discriminator attribute)
(GAN_CIFAR10_jax.Discriminator_cond attribute)
(GAN_CIFAR10_jax.Generator attribute)
(GAN_CIFAR10_jax.Generator_cond attribute)
(GAN_MNIST_jax.Discriminator_MNIST_cond attribute)
(GAN_MNIST_jax.Generator_MNIST_cond attribute)
(model_jax.Discriminator attribute)
(model_jax.DiscriminatorMNIST attribute)
(model_jax.Generator attribute)
set_batch_size() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
set_discriminator() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
set_learning_rate() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
set_Lip_constant() (Divergences_jax.Gradient_Penalty_1Sided method)
,
[1]
(Divergences_jax.Gradient_Penalty_2Sided method)
,
[1]
(Divergences_tf.Gradient_Penalty_1Sided method)
,
[1]
(Divergences_tf.Gradient_Penalty_2Sided method)
,
[1]
(Divergences_torch.Gradient_Penalty_1Sided method)
,
[1]
(Divergences_torch.Gradient_Penalty_2Sided method)
,
[1]
set_no_epochs() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
set_order() (Divergences_jax.alpha_Divergence_LT method)
,
[1]
(Divergences_jax.Renyi_Divergence method)
,
[1]
(Divergences_tf.alpha_Divergence_LT method)
,
[1]
(Divergences_tf.Renyi_Divergence method)
,
[1]
(Divergences_torch.alpha_Divergence_LT method)
,
[1]
(Divergences_torch.Renyi_Divergence method)
,
[1]
set_penalty_weight() (Divergences_jax.Discriminator_Penalty method)
,
[1]
(Divergences_tf.Discriminator_Penalty method)
,
[1]
(Divergences_torch.Discriminator_Penalty method)
,
[1]
spec_norm (model_jax.Discriminator attribute)
(model_jax.Generator attribute)
squared_Hellinger_LT (class in Divergences_jax)
,
[1]
(class in Divergences_tf)
,
[1]
(class in Divergences_torch)
,
[1]
T
train() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
(GAN_jax.GAN method)
,
[1]
(GAN_tf.GAN method)
,
[1]
(GAN_torch.GAN method)
,
[1]
train_step() (Divergences_jax.Divergence method)
,
[1]
(Divergences_tf.Divergence method)
,
[1]
(Divergences_torch.Divergence method)
,
[1]
X
X_dim (model_jax.Generator attribute)
Z
Z_dim (model_jax.Generator attribute)