navigation links

Index

A

AdaptiveSurveyIntegrator (class in zunis.integration.adaptive_survey_integrator) AnalyticFlow (class in zunis.models.flows.analytic_flows.analytic_flow) AnalyticInverseSigmoid (class in zunis.models.flows.analytic_flows.element_wise) AnalyticSigmoid (class in zunis.models.flows.analytic_flows.element_wise)

append_dataframe_to_sqlite() (in module utils.data_storage.dataframe2sql) append_record_to_sqlite() (in module utils.data_storage.record2sql) ArbitraryShapeRectangularDNN (class in zunis.models.layers.trainable) AvertedCUDARuntimeError

B

backward() (ElementWiseAffineTransform method) (ElementWisePWLinearTransform static method) (ElementWisePWQuadraticTransform static method) (InvertibleTransform method) BaseIntegrator (class in zunis.integration.base_integrator) BasicDKLTrainer (class in zunis.training.weighted_dataset.dkl_training) BasicStatefulDKLTrainer (class in zunis.training.weighted_dataset.dkl_training) BasicStatefulTrainer (class in zunis.training.weighted_dataset.weighted_dataset_trainer)

BasicStatefulVarTrainer (class in zunis.training.weighted_dataset.variance_training) BasicTrainer (class in zunis.training.weighted_dataset.weighted_dataset_trainer) BasicVarTrainer (class in zunis.training.weighted_dataset.variance_training) benchmark_method() (Benchmarker method) (KnownIntegrandBenchmarker method) (VegasBenchmarker method) Benchmarker (class in utils.benchmark.benchmarker) BiLU (class in zunis.models.layers.activations)

C

CamelIntegrand (class in utils.integrands.camel) cells (RepeatedCellFlow attribute) check() (KnownIntegrand method) CircleLineIntegrand (class in utils.integrands.pretty) compare_absolute() (KnownIntegrand method) compare_integral_result() (in module utils.integral_validation) compare_integrals() (in module utils.integral_validation) compare_relative() (KnownIntegrand method) ComparisonRecord (class in utils.record) compute_flat_variance_loss() (VarianceAdaptiveSurveyIntegrator method) compute_loss_no_grad() (BasicTrainer method)

compute_mc_statistics() (in module utils.integrals) Configuration (class in zunis.utils.config.configuration) create_checkerboard_hypercube_realnvp() (in module zunis.models.generators) create_dkl_trainer() (in module zunis.training.weighted_dataset.generators) create_hypercube_flow() (in module zunis.models.generators) create_integrator_args() (in module zunis.utils.config.loaders) create_integrator_config_file() (in module zunis.utils.config.generators) create_n_ary_hypercube_flow() (in module zunis.models.generators) create_n_ary_hypercube_realnvp() (in module zunis.models.generators) create_rectangular_dnn() (in module zunis.models.layers.trainable) CrossSection (class in utils.integrands.madgraph)

D

default_flow_priors (StatefulTrainer attribute)

DiagonalGaussianIntegrand (class in utils.integrands.gaussian) DKLAdaptiveSurveyIntegrator (class in zunis.integration.dkltrainer_integrator)

E

element_wise_affine() (in module zunis.models.flows.coupling_cells.real_nvp) ElementWiseAffineTransform (class in zunis.models.flows.coupling_cells.real_nvp) ElementWisePWLinearTransform (class in zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_linear) ElementWisePWQuadraticTransform (class in zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_quadratic) empty_history() (BaseIntegrator static method) epoch (TrainingRecord property) erf() (KnownGaussianIntegrand static method) evaluate_integral() (in module utils.integral_validation) evaluate_integral_flat() (in module utils.flat_integrals) evaluate_integral_integrator() (in module utils.integrator_integrals) evaluate_integral_stratified() (in module utils.integral_validation) evaluate_integral_vegas() (in module utils.vegas_integrals)

evaluate_integrand() (CamelIntegrand method) (CircleLineIntegrand method) (CrossSection method) (DiagonalGaussianIntegrand method) (Integrand method) (RegulatedHyperSphericalCamel method) (RegulatedIntegrand method) (SineIntegrand method) (SineLineIntegrand method) (VolumeIntegrand method) evaluate_known_integral() (in module utils.known_integrals) EvaluationRecord (class in utils.record)

F

FactorizedFlowSampler (class in zunis.models.flows.sampling) FactorizedGaussianSampler (class in zunis.models.flows.sampling) FakeKnownSampler (class in utils.known_integrals) FakeRealNVP (class in zunis.models.flows.coupling_cells.real_nvp) FakeT (class in zunis.models.flows.coupling_cells.real_nvp) finalize_integration() (BaseIntegrator method) (SurveyRefineIntegratorAPI method) finalize_refine() (BaseIntegrator method) (SurveyRefineIntegratorAPI method) finalize_survey() (BaseIntegrator method) (SurveyRefineIntegratorAPI method) FixedSampleSurveyIntegrator (class in zunis.integration.fixed_sample_integrator) FlatSampler (class in utils.flat_integrals) FlatSurveySamplingIntegrator (class in zunis.integration.flat_survey_integrator) flow() (AnalyticFlow method) (GeneralBackpropJacobianFlow method) (GeneralCouplingCell method) (GeneralFlow method) (InvertibleAnalyticFlow method) (InvertibleSequentialFlow method)

flow_priors (StatefulTrainer attribute) format_arguments() (SurveyRefineIntegratorAPI method) forward() (ArbitraryShapeRectangularDNN method) (BiLU method) (ElementWiseAffineTransform method) (ElementWisePWLinearTransform static method) (ElementWisePWQuadraticTransform static method) (FactorizedFlowSampler method) (FakeT method) (GeneralFlow method) (InvertibleTransform method) (NormBiTanh method) (OverallAffineLayer method) (Reshift method) ForwardSurveySamplingIntegrator (class in zunis.integration.adaptive_survey_integrator)

G

GeneralBackpropJacobianFlow (class in zunis.models.flows.backprop_jacobian_flows.general_backprop_j_flow) GeneralCouplingCell (class in zunis.models.flows.coupling_cells.general_coupling) GeneralFlow (class in zunis.models.flows.general_flow) GeneralPWLinearCoupling (class in zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_linear) GeneralPWQuadraticCoupling (class in zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_quadratic) GeneralRealNVP (class in zunis.models.flows.coupling_cells.real_nvp) generate_config_samples() (Benchmarker method) (GridBenchmarker method) (RandomHyperparameterBenchmarker method) (SequentialBenchmarker method) (SequentialIntegratorBenchmarker method) generate_target_batch_from_posterior() (BasicTrainer static method) GenericTrainerAPI (class in zunis.training.weighted_dataset.weighted_dataset_trainer) get_bin() (in module zunis.models.flows.masking)

get_config() (BasicStatefulTrainer method) (GenericTrainerAPI method) get_default_integrator_config() (in module zunis.utils.config.loaders) get_device() (in module utils.torch_utils) get_dirt_hash() (in module utils.repo_info) get_git_summary() (in module utils.repo_info) get_history() (IntegratorSampler method) (Sampler method) get_optim_from_config() (in module zunis.utils.config.loaders) get_pipfreeze_hash() (in module utils.repo_info) get_repo() (in module utils.repo_info) get_repo_hash() (in module utils.repo_info) get_sql_types() (in module utils.config.loaders) GridBenchmarker (class in utils.benchmark.benchmarker)

H

handle_cuda_error() (BasicStatefulTrainer method) (BasicTrainer method) handle_invalid_loss() (BasicStatefulTrainer method) (BasicTrainer method)

HyperrectangleVolumeIntegrand (class in utils.integrands.volume) HypersphereVolumeIntegrand (class in utils.integrands.volume)

I

iflow_strategy() (in module zunis.models.flows.masking) inequality() (HyperrectangleVolumeIntegrand method) (HypersphereVolumeIntegrand method) (VolumeIntegrand method) infer_sql_types() (in module utils.data_storage.record2sql) initialize() (BaseIntegrator method) (SurveyRefineIntegratorAPI method) initialize_refine() (BaseIntegrator method) (SurveyRefineIntegratorAPI method) initialize_survey() (BaseIntegrator method) (SurveyRefineIntegratorAPI method) integral() (HyperrectangleVolumeIntegrand method) (HypersphereVolumeIntegrand method) (KnownGaussianIntegrand method) (KnownIntegrand method) (KnownSymmetricCamelIntegrand method) (RegulatedHyperSphericalCamel method) (RegulatedKnownIntegrand method)

Integrand (class in utils.integrands.abstract) integrate() (BaseIntegrator method) (SurveyRefineIntegratorAPI method) integration_logger (in module zunis.integration) Integrator() (in module zunis.integration.default_integrator) IntegratorSampler (class in utils.integrator_integrals) InvalidLossError inverse (InvertibleTransform attribute) inverse_element_wise_affine() (in module zunis.models.flows.coupling_cells.real_nvp) inverse_sigmoid() (in module zunis.models.flows.analytic_flows.element_wise) invert() (InvertibleAnalyticFlow method) (InvertibleCouplingCell method) (InvertibleSequentialFlow method) (InvertibleTransform method) InvertibleAnalyticFlow (class in zunis.models.flows.analytic_flows.analytic_flow) InvertibleAnalyticSigmoid (class in zunis.models.flows.analytic_flows.element_wise) InvertibleCouplingCell (class in zunis.models.flows.coupling_cells.general_coupling) InvertibleSequentialFlow (class in zunis.models.flows.sequential.invertible_sequential) InvertibleTransform (class in zunis.models.flows.coupling_cells.transforms)

K

KnownGaussianIntegrand (class in utils.integrands.gaussian) KnownIntegrand (class in utils.integrands.abstract) KnownIntegrandBenchmarker (class in utils.benchmark.known_integrand_benchmarks)

KnownIntegrandGridBenchmarker (class in utils.benchmark.known_integrand_benchmarks) KnownIntegrandRandomHPBenchmarker (class in utils.benchmark.known_integrand_benchmarks) KnownSymmetricCamelIntegrand (class in utils.integrands.camel)

L

LinearFlow (class in zunis.models.flows.backprop_jacobian_flows.simple_backprop_flows) log_loss() (TrainingRecord method) log_metric() (TrainingRecord method) log_prob() (FactorizedFlowSampler method) logger (in module zunis)

logger_integration (in module zunis) logger_training (in module zunis) loss (TrainingRecord property) loss_map (StatefulTrainer attribute) losses (TrainingRecord property)

M

masking (RepeatedCellFlow attribute) MaskListRepeatedCellFlow (class in zunis.models.flows.sequential.repeated_cell) maximal_masking_strategy() (in module zunis.models.flows.masking) metrics (TrainingRecord property) modified_softmax() (in module zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_quadratic) module utils utils.benchmark utils.benchmark.benchmark_time utils.benchmark.benchmarker utils.benchmark.known_integrand_benchmarks utils.benchmark.vegas_benchmarks utils.command_line_tools utils.config utils.config.loaders utils.data_storage utils.data_storage.dataframe2sql utils.data_storage.record2sql utils.flat_integrals utils.integral_validation utils.integrals utils.integrands utils.integrands.abstract utils.integrands.camel utils.integrands.gaussian utils.integrands.madgraph utils.integrands.pretty utils.integrands.volume utils.integrator_integrals utils.known_integrals utils.logging utils.record utils.repo_info utils.torch_utils utils.vanity utils.vegas_integrals zunis zunis.integration zunis.integration.adaptive_survey_integrator zunis.integration.base_integrator zunis.integration.default_integrator zunis.integration.dkltrainer_integrator zunis.integration.fixed_sample_integrator zunis.integration.flat_survey_integrator zunis.integration.integratorAPI zunis.models zunis.models.flows zunis.models.flows.analytic_flows zunis.models.flows.analytic_flows.analytic_flow zunis.models.flows.analytic_flows.element_wise zunis.models.flows.backprop_jacobian_flows zunis.models.flows.backprop_jacobian_flows.general_backprop_j_flow zunis.models.flows.backprop_jacobian_flows.nnflows zunis.models.flows.backprop_jacobian_flows.simple_backprop_flows zunis.models.flows.coupling_cells zunis.models.flows.coupling_cells.general_coupling zunis.models.flows.coupling_cells.piecewise_coupling zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_linear zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_quadratic zunis.models.flows.coupling_cells.real_nvp zunis.models.flows.coupling_cells.transforms zunis.models.flows.general_flow zunis.models.flows.masking zunis.models.flows.sampling zunis.models.flows.sequential zunis.models.flows.sequential.invertible_sequential zunis.models.flows.sequential.repeated_cell zunis.models.generators zunis.models.layers zunis.models.layers.activations zunis.models.layers.trainable zunis.models.utils zunis.training zunis.training.weighted_dataset zunis.training.weighted_dataset.dkl_training zunis.training.weighted_dataset.generators zunis.training.weighted_dataset.stateful_trainer zunis.training.weighted_dataset.training_record zunis.training.weighted_dataset.variance_training zunis.training.weighted_dataset.weighted_dataset_trainer zunis.utils zunis.utils.config zunis.utils.config.configuration zunis.utils.config.defaults zunis.utils.config.generators zunis.utils.config.loaders zunis.utils.exceptions zunis.utils.function_wrapper zunis.utils.logger

N

n_ary_mask() (in module zunis.models.flows.masking) n_ary_mask_strategy() (in module zunis.models.flows.masking) new_epoch() (TrainingRecord method) next_step() (TrainingRecord method)

NNFlow (class in zunis.models.flows.backprop_jacobian_flows.nnflows) NoCheckpoint NormBiTanh (class in zunis.models.layers.activations) not_list() (in module zunis.models.flows.coupling_cells.general_coupling)

O

OverallAffineLayer (class in zunis.models.layers.trainable)

P

piecewise_linear_inverse_transform() (in module zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_linear) piecewise_linear_transform() (in module zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_linear) piecewise_quadratic_inverse_transform() (in module zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_quadratic) piecewise_quadratic_transform() (in module zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_quadratic) PosteriorSurveySamplingIntegrator (class in zunis.integration.flat_survey_integrator) process_loss() (BasicStatefulTrainer method) (BasicTrainer method) process_refine_step() (BaseIntegrator method) (SurveyRefineIntegratorAPI method)

process_survey_step() (AdaptiveSurveyIntegrator method) (BaseIntegrator method) (SurveyRefineIntegratorAPI method) (VarianceAdaptiveSurveyIntegrator method) PWLinearCoupling (class in zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_linear) PWQuadraticCoupling (class in zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_quadratic) PythonLiteralOption (class in utils.command_line_tools)

R

RandomHyperparameterBenchmarker (class in utils.benchmark.benchmarker) read_pkl_sql() (in module utils.data_storage.dataframe2sql) RealNVP (class in zunis.models.flows.coupling_cells.real_nvp) Record (class in utils.record) refine() (BaseIntegrator method) (SurveyRefineIntegratorAPI method) refine_step() (SurveyRefineIntegratorAPI method) RegulatedDiagonalGaussianIntegrand (class in utils.integrands.gaussian) RegulatedHyperSphereIntegrand (class in utils.integrands.volume) RegulatedHyperSphericalCamel (class in utils.integrands.volume) RegulatedIntegrand (class in utils.integrands.abstract)

RegulatedKnownIntegrand (class in utils.integrands.abstract) RegulatedSymmetricHyperSphericalCamel (class in utils.integrands.volume) RepeatedCellFlow (class in zunis.models.flows.sequential.repeated_cell) repo_is_dirty() (in module utils.repo_info) reset() (GenericTrainerAPI method) Reshift (class in zunis.models.utils) restore_checkpoint() (BasicStatefulTrainer method) restore_checkpoint_except() (BasicStatefulTrainer method) run() (Benchmarker method) run_time_benchmark() (in module utils.benchmark.benchmark_time) runs_forward() (InvertibleAnalyticFlow method) (InvertibleCouplingCell method)

S

sample() (FakeKnownSampler method) (FlatSampler method) (IntegratorSampler method) (Sampler method) (VegasSampler method) sample_forward() (BasicTrainer method) (GenericTrainerAPI method) sample_non_stratified() (VegasSampler method) sample_refine() (BaseIntegrator method) (SurveyRefineIntegratorAPI method) sample_stratified() (VegasSampler method) sample_survey() (AdaptiveSurveyIntegrator method) (FixedSampleSurveyIntegrator method) (PosteriorSurveySamplingIntegrator method) (SurveyRefineIntegratorAPI method) Sampler (class in utils.integral_validation) sanitize_variable() (in module utils.integrands) SequentialBenchmarker (class in utils.benchmark.benchmarker) SequentialIntegratorBenchmarker (class in utils.benchmark.benchmarker) set_benchmark_grid_config() (Benchmarker method) set_benchmark_grid_config_param() (Benchmarker static method) set_benchmark_logger() (in module utils.logging) set_benchmark_logger_debug() (in module utils.logging)

set_checkpoint() (BasicStatefulTrainer method) set_config() (BasicStatefulTrainer method) (GenericTrainerAPI method) set_sample() (FixedSampleSurveyIntegrator method) set_sample_csv() (FixedSampleSurveyIntegrator method) set_sample_pickle() (FixedSampleSurveyIntegrator method) set_verbosity() (BasicTrainer method) (in module zunis.utils.logger) (SurveyRefineIntegratorAPI method) setup_std_stream_logger() (in module zunis) SigmoidFlow (class in zunis.models.flows.backprop_jacobian_flows.simple_backprop_flows) SineIntegrand (class in utils.integrands.pretty) SineLineIntegrand (class in utils.integrands.pretty) StatefulTrainer (class in zunis.training.weighted_dataset.stateful_trainer) step (TrainingRecord property) survey() (BaseIntegrator method) (SurveyRefineIntegratorAPI method) survey_step() (SurveyRefineIntegratorAPI method) survey_switch_condition() (AdaptiveSurveyIntegrator method) (DKLAdaptiveSurveyIntegrator method) (ForwardSurveySamplingIntegrator method) (VarianceAdaptiveSurveyIntegrator method) SurveyRefineIntegratorAPI (class in zunis.integration.integratorAPI) SymmetricCamelIntegrand (class in utils.integrands.camel)

T

train_integrator() (IntegratorSampler method) (VegasSampler method) train_on_batch() (BasicStatefulTrainer method) (GenericTrainerAPI method) train_on_target_batch() (BasicTrainer method) train_on_target_batches_from_posterior() (BasicStatefulTrainer method) (BasicTrainer method) train_step_on_target_batch() (BasicStatefulTrainer method) (BasicTrainer method) train_step_on_target_minibatch() (BasicStatefulTrainer method) (BasicTrainer method) training (AnalyticFlow attribute) (AnalyticInverseSigmoid attribute) (AnalyticSigmoid attribute) (ArbitraryShapeRectangularDNN attribute) (BiLU attribute) (FactorizedFlowSampler attribute) (FactorizedGaussianSampler attribute) (FakeRealNVP attribute) (FakeT attribute) (GeneralBackpropJacobianFlow attribute) (GeneralCouplingCell attribute) (GeneralFlow attribute) (GeneralPWLinearCoupling attribute) (GeneralPWQuadraticCoupling attribute) (GeneralRealNVP attribute) (InvertibleAnalyticFlow attribute) (InvertibleAnalyticSigmoid attribute) (InvertibleCouplingCell attribute) (InvertibleSequentialFlow attribute) (LinearFlow attribute) (MaskListRepeatedCellFlow attribute) (NNFlow attribute) (NormBiTanh attribute) (OverallAffineLayer attribute) (PWLinearCoupling attribute) (PWQuadraticCoupling attribute) (RealNVP attribute) (RepeatedCellFlow attribute) (Reshift attribute) (SigmoidFlow attribute) (UniformSampler attribute)

training_logger (in module zunis.training) TrainingInterruption TrainingRecord (class in zunis.training.weighted_dataset.training_record) transform_and_compute_jacobian() (AnalyticInverseSigmoid method) (AnalyticSigmoid method) (GeneralBackpropJacobianFlow method) (GeneralCouplingCell method) (GeneralFlow method) (InvertibleAnalyticFlow method) (InvertibleSequentialFlow method) type_cast_value() (PythonLiteralOption method)

U

UniformSampler (class in zunis.models.flows.sampling) utils module utils.benchmark module utils.benchmark.benchmark_time module utils.benchmark.benchmarker module utils.benchmark.known_integrand_benchmarks module utils.benchmark.vegas_benchmarks module utils.command_line_tools module utils.config module utils.config.loaders module utils.data_storage module utils.data_storage.dataframe2sql module utils.data_storage.record2sql module utils.flat_integrals module utils.integral_validation module utils.integrals module

utils.integrands module utils.integrands.abstract module utils.integrands.camel module utils.integrands.gaussian module utils.integrands.madgraph module utils.integrands.pretty module utils.integrands.volume module utils.integrator_integrals module utils.known_integrals module utils.logging module utils.record module utils.repo_info module utils.torch_utils module utils.vanity module utils.vegas_integrals module

V

validate_integral() (in module utils.integral_validation) validate_integral_integrator() (in module utils.integrator_integrals) validate_known_integrand_flat() (in module utils.flat_integrals) VarianceAdaptiveSurveyIntegrator (class in zunis.integration.adaptive_survey_integrator) vegas() (Integrand method) VegasBenchmarker (class in utils.benchmark.vegas_benchmarks)

VegasGridBenchmarker (class in utils.benchmark.vegas_benchmarks) VegasRandomHPBenchmarker (class in utils.benchmark.vegas_benchmarks) VegasSampler (class in utils.vegas_integrals) VegasSequentialBenchmarker (class in utils.benchmark.vegas_benchmarks) VegasSequentialIntegratorBenchmarker (class in utils.benchmark.vegas_benchmarks) VolumeIntegrand (class in utils.integrands.volume)

W

weight_init_identity_() (LinearFlow method) (NNFlow method) weighted_dkl_loss() (in module zunis.training.weighted_dataset.dkl_training) weighted_variance_loss() (in module zunis.training.weighted_dataset.variance_training)

wrap_compact_arguments_function() (in module zunis.utils.function_wrapper) wrap_hypercube_arguments_function() (in module zunis.utils.function_wrapper) wrap_numpy_compact_batch_function() (in module zunis.utils.function_wrapper) wrap_numpy_hypercube_batch_function() (in module zunis.utils.function_wrapper)

Z

zunis module zunis.integration module zunis.integration.adaptive_survey_integrator module zunis.integration.base_integrator module zunis.integration.default_integrator module zunis.integration.dkltrainer_integrator module zunis.integration.fixed_sample_integrator module zunis.integration.flat_survey_integrator module zunis.integration.integratorAPI module zunis.models module zunis.models.flows module zunis.models.flows.analytic_flows module zunis.models.flows.analytic_flows.analytic_flow module zunis.models.flows.analytic_flows.element_wise module zunis.models.flows.backprop_jacobian_flows module zunis.models.flows.backprop_jacobian_flows.general_backprop_j_flow module zunis.models.flows.backprop_jacobian_flows.nnflows module zunis.models.flows.backprop_jacobian_flows.simple_backprop_flows module zunis.models.flows.coupling_cells module zunis.models.flows.coupling_cells.general_coupling module zunis.models.flows.coupling_cells.piecewise_coupling module zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_linear module zunis.models.flows.coupling_cells.piecewise_coupling.piecewise_quadratic module zunis.models.flows.coupling_cells.real_nvp module zunis.models.flows.coupling_cells.transforms module zunis.models.flows.general_flow module zunis.models.flows.masking module

zunis.models.flows.sampling module zunis.models.flows.sequential module zunis.models.flows.sequential.invertible_sequential module zunis.models.flows.sequential.repeated_cell module zunis.models.generators module zunis.models.layers module zunis.models.layers.activations module zunis.models.layers.trainable module zunis.models.utils module zunis.training module zunis.training.weighted_dataset module zunis.training.weighted_dataset.dkl_training module zunis.training.weighted_dataset.generators module zunis.training.weighted_dataset.stateful_trainer module zunis.training.weighted_dataset.training_record module zunis.training.weighted_dataset.variance_training module zunis.training.weighted_dataset.weighted_dataset_trainer module zunis.utils module zunis.utils.config module zunis.utils.config.configuration module zunis.utils.config.defaults module zunis.utils.config.generators module zunis.utils.config.loaders module zunis.utils.exceptions module zunis.utils.function_wrapper module zunis.utils.logger module