Index

A

AccumulationSampler (class in samplers.accumulation_sampler) add() (utils.moving_average.MovingAverage method)

add_array() (utils.moving_average.MovingAverage method) average() (in module utils.moving_average)

B

BaseDistribution (class in distributions.base_distribution) BaseDivergence (class in metrics.base) BaseLoss (class in tuners.losses.base)

BaseTunerObserver (class in tuners.loggers.base) batchify() (in module utils.helpers) BooleanScorer (class in scorers.boolean_scorer)

C

CDPGTuner (class in tuners.cdpg_tuner) ConsoleLogger (class in tuners.loggers.console)

constrain() (distributions.base_distribution.BaseDistribution method) ContextDistribution (class in distributions.context_distribution)

D

DatasetContextDistribution (class in distributions.dataset_context_distribution) default_params (tuners.tuner.Tuner attribute) dispatch() (utils.observable.Observable method) Distribution (class in distributions.distribution) distributions module distributions.base_distribution module distributions.context_distribution module

distributions.dataset_context_distribution module distributions.distribution module distributions.lm_distribution module distributions.single_context_distribution module divergence() (metrics.base.BaseDivergence class method) DPGTuner (class in tuners.dpg_tuner)

E

enroll() (utils.observable.Observable method)

ExponentialScorer (class in scorers.exponential_scorer)

F

forward() (in module utils.observable)

freeze() (distributions.lm_distribution.LMDistribution method)

G

get_acceptance_rate() (samplers.quasi_rejection_sampler.QuasiRejectionSampler method) get_device() (in module utils.device)

get_proxies() (in module tuners.loggers.neptune) get_token_first_indices() (in module utils.helpers)

J

JS (class in metrics.js)

JSONLogger (class in tuners.loggers.json)

K

KL (class in metrics.kl)

KLLoss (class in tuners.losses.kl)

L

LMDistribution (class in distributions.lm_distribution) log_score() (distributions.context_distribution.ContextDistribution method) (distributions.dataset_context_distribution.DatasetContextDistribution method) (distributions.lm_distribution.LMDistribution method) (distributions.single_context_distribution.SingleContextDistribution method) (scorers.boolean_scorer.BooleanScorer method) (scorers.exponential_scorer.ExponentialScorer method) (scorers.pipeline_scorer.PipelineScorer method) (scorers.positive_scorer.PositiveScorer method) (scorers.positive_scorer.Product method)

M

metrics module metrics.base module metrics.js module metrics.kl module metrics.tv module module distributions distributions.base_distribution distributions.context_distribution distributions.dataset_context_distribution distributions.distribution distributions.lm_distribution distributions.single_context_distribution metrics metrics.base metrics.js metrics.kl metrics.tv samplers samplers.accumulation_sampler samplers.quasi_rejection_sampler samplers.sampler scorers scorers.boolean_scorer scorers.exponential_scorer scorers.pipeline_scorer scorers.positive_scorer scorers.scorer tuners tuners.cdpg_tuner tuners.dpg_tuner tuners.loggers tuners.loggers.base tuners.loggers.console tuners.loggers.json tuners.loggers.neptune tuners.loggers.wandb tuners.losses tuners.losses.base tuners.losses.kl tuners.tuner utils utils.device utils.helpers utils.moving_average utils.observable

MovingAverage (class in utils.moving_average)

N

NeptuneLogger (class in tuners.loggers.neptune)

O

Observable (class in utils.observable) on_eval_samples_updated() (tuners.loggers.base.BaseTunerObserver method) (tuners.loggers.json.JSONLogger method) (tuners.loggers.neptune.NeptuneLogger method) (tuners.loggers.wandb.WandBLogger method) on_metric_updated() (tuners.loggers.base.BaseTunerObserver method) (tuners.loggers.json.JSONLogger method) (tuners.loggers.neptune.NeptuneLogger method) (tuners.loggers.wandb.WandBLogger method) on_ministep_idx_updated() (tuners.loggers.base.BaseTunerObserver method) (tuners.loggers.console.ConsoleLogger method) (tuners.loggers.json.JSONLogger method) (tuners.loggers.neptune.NeptuneLogger method) (tuners.loggers.wandb.WandBLogger method)

on_parameters_updated() (tuners.loggers.base.BaseTunerObserver method) (tuners.loggers.console.ConsoleLogger method) (tuners.loggers.json.JSONLogger method) (tuners.loggers.neptune.NeptuneLogger method) (tuners.loggers.wandb.WandBLogger method) on_step_idx_updated() (tuners.loggers.base.BaseTunerObserver method) (tuners.loggers.console.ConsoleLogger method) (tuners.loggers.json.JSONLogger method) (tuners.loggers.neptune.NeptuneLogger method) (tuners.loggers.wandb.WandBLogger method)

P

PipelineScorer (class in scorers.pipeline_scorer) pointwise_estimates() (metrics.js.JS class method) (metrics.kl.KL class method) (metrics.tv.TV class method)

PositiveScorer (class in scorers.positive_scorer) Product (class in scorers.positive_scorer)

Q

QuasiRejectionSampler (class in samplers.quasi_rejection_sampler)

R

report_feature_moments() (tuners.tuner.Tuner method)

reset() (utils.moving_average.MovingAverage method)

S

sample() (distributions.context_distribution.ContextDistribution method) (distributions.dataset_context_distribution.DatasetContextDistribution method) (distributions.distribution.Distribution method) (distributions.lm_distribution.LMDistribution method) (distributions.single_context_distribution.SingleContextDistribution method) (samplers.accumulation_sampler.AccumulationSampler method) (samplers.quasi_rejection_sampler.QuasiRejectionSampler method) (samplers.sampler.Sampler method) Sampler (class in samplers.sampler) samplers module samplers.accumulation_sampler module samplers.quasi_rejection_sampler module samplers.sampler module

save() (tuners.loggers.json.JSONLogger method) score() (scorers.boolean_scorer.BooleanScorer method) (scorers.positive_scorer.PositiveScorer method) (scorers.scorer.Scorer method) Scorer (class in scorers.scorer) scorers module scorers.boolean_scorer module scorers.exponential_scorer module scorers.pipeline_scorer module scorers.positive_scorer module scorers.scorer module SingleContextDistribution (class in distributions.single_context_distribution)

T

text (distributions.lm_distribution.TextSample property) TextSample (class in distributions.lm_distribution) to() (distributions.lm_distribution.LMDistribution method) to_same_device() (in module utils.device) token_ids (distributions.lm_distribution.TextSample property) tune() (tuners.tuner.Tuner method) Tuner (class in tuners.tuner) tuners module tuners.cdpg_tuner module tuners.dpg_tuner module tuners.loggers module tuners.loggers.base module

tuners.loggers.console module tuners.loggers.json module tuners.loggers.neptune module tuners.loggers.wandb module tuners.losses module tuners.losses.base module tuners.losses.kl module tuners.tuner module TV (class in metrics.tv)

U

utils module utils.device module utils.helpers module

utils.moving_average module utils.observable module

W

WandBLogger (class in tuners.loggers.wandb)