Index

_

__call__() (kiwi.lib.search.Objective method) __contains__() (kiwi.data.datasets.parallel_dataset.ParallelDataset method) (kiwi.data.datasets.wmt_qe_dataset.WMTQEDataset method) __copyright__ (in module kiwi) __getitem__() (kiwi.data.datasets.parallel_dataset.ParallelDataset method) (kiwi.data.datasets.wmt_qe_dataset.WMTQEDataset method) (kiwi.utils.data_structures.DefaultFrozenDict method) __getstate__() (kiwi.data.vocabulary.Vocabulary method) __len__() (kiwi.data.datasets.parallel_dataset.ParallelDataset method) (kiwi.data.datasets.wmt_qe_dataset.WMTQEDataset method) (kiwi.data.vocabulary.Vocabulary method) __setstate__() (kiwi.data.vocabulary.Vocabulary method) __str__() (kiwi.lib.evaluate.MetricsReport method) __version__ (in module kiwi) _check_freezing() (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder method) _extract_path_prefix() (in module kiwi.lib.evaluate) _INVALID_PARAM_AND_METRIC_CHARACTERS (in module kiwi.loggers)

_load_dict() (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) _load_encoder() (kiwi.systems.qe_system.QESystem method) _loss_fn() (kiwi.modules.sentence_level_output.SentenceScoreDistribution static method) _name (kiwi.metrics.metrics.CorrectMetric attribute) (kiwi.metrics.metrics.ExpectedErrorMetric attribute) (kiwi.metrics.metrics.F1MultMetric attribute) (kiwi.metrics.metrics.MatthewsMetric attribute) (kiwi.metrics.metrics.Metric attribute) (kiwi.metrics.metrics.NLLMetric attribute) (kiwi.metrics.metrics.PearsonMetric attribute) (kiwi.metrics.metrics.PerplexityMetric attribute) (kiwi.metrics.metrics.RMSEMetric attribute) (kiwi.metrics.metrics.SpearmanMetric attribute) _reset_hydra() (in module kiwi.lib.utils) _reverse_padded_seq() (kiwi.systems.encoders.predictor.DualSequencesEncoder static method) _scores_str() (kiwi.lib.evaluate.MetricsReport static method) _split_hidden() (kiwi.systems.encoders.predictor.DualSequencesEncoder static method)

A

add_embeddings_vocab (kiwi.data.encoders.parallel_data_encoder.VocabularyConfig attribute) (kiwi.data.encoders.wmt_qe_data_encoder.VocabularyConfig attribute) add_sentence_level_scores() (kiwi.lib.evaluate.MetricsReport method) add_word_level_scores() (kiwi.lib.evaluate.MetricsReport method) align_tokenize() (in module kiwi.data.tokenizers) AlignmentEncoder (class in kiwi.data.encoders.field_encoders)

ALIGNMENTS (in module kiwi.constants) alignments (kiwi.data.datasets.wmt_qe_dataset.InputConfig attribute) amp_level (kiwi.lib.train.GPUConfig attribute) apply_packed_sequence() (in module kiwi.utils.tensors) arg_constraints (kiwi.modules.common.distributions.TruncatedNormal attribute) arguments_to_configuration() (in module kiwi.lib.utils) Attention (class in kiwi.modules.common.attention)

B

backward() (kiwi.utils.tensors.GradientMul static method) BAD (in module kiwi.constants) base_config (kiwi.lib.search.Configuration attribute) BaseConfig (class in kiwi.utils.io) BaseConfig.Config (class in kiwi.utils.io) batch_encode() (kiwi.data.encoders.field_encoders.AlignmentEncoder method) (kiwi.data.encoders.field_encoders.ScoreEncoder method) (kiwi.data.encoders.field_encoders.TextEncoder method) batch_size (in module kiwi.modules.common.positional_encoding) (kiwi.systems.qe_system.QESystem.Config attribute) (kiwi.systems.tlm_system.TLMSystem.Config attribute) BatchedSentence (class in kiwi.data.batch) BatchSizeConfig (class in kiwi.systems.qe_system) (class in kiwi.systems.tlm_system) Bert (class in kiwi.systems.bert) Bert.Config (class in kiwi.systems.bert) bert_tokenizer() (in module kiwi.data.tokenizers) BertEncoder (class in kiwi.systems.encoders.bert) BertEncoder.Config (class in kiwi.systems.encoders.bert) best_metrics (kiwi.lib.train.TrainRunInfo attribute) best_model_path (kiwi.lib.train.TrainRunInfo attribute)

best_model_paths() (kiwi.lib.search.Objective property) best_ordering (kiwi.metrics.metrics.ExpectedErrorMetric attribute) (kiwi.metrics.metrics.Metric attribute) (kiwi.metrics.metrics.NLLMetric attribute) (kiwi.metrics.metrics.PerplexityMetric attribute) (kiwi.metrics.metrics.RMSEMetric attribute) best_train_configs() (kiwi.lib.search.Objective property) BestMetricsInfo (class in kiwi.training.callbacks) BINARY (in module kiwi.constants) binary (kiwi.systems.outputs.quality_estimation.SentenceLevelConfig attribute) BinaryScoreEncoder (class in kiwi.data.encoders.field_encoders) BinarySentenceScore (class in kiwi.modules.sentence_level_output) bos_id() (kiwi.data.vocabulary.Vocabulary property) bottleneck_size (kiwi.lib.search.SearchOptions attribute) (kiwi.systems.decoders.linear.LinearDecoder.Config attribute) bounds (kiwi.data.batch.BatchedSentence attribute) bounds_lengths (kiwi.data.batch.BatchedSentence attribute) buffer_size (kiwi.data.datasets.parallel_dataset.ParallelDataset.Config attribute) (kiwi.data.datasets.wmt_qe_dataset.WMTQEDataset.Config attribute) build() (kiwi.data.datasets.parallel_dataset.ParallelDataset static method) (kiwi.data.datasets.wmt_qe_dataset.WMTQEDataset static method)

C

cast_steps() (kiwi.training.optimizers.OptimizerConfig method) cdf() (kiwi.modules.common.distributions.TruncatedNormal method) check_batching() (kiwi.systems.qe_system.QESystem.Config method) (kiwi.systems.tlm_system.TLMSystem.Config method) check_consistency() (kiwi.lib.evaluate.Configuration method) (kiwi.lib.predict.RunConfig method) (kiwi.lib.search.SearchOptions method) (kiwi.systems.qe_system.QESystem.Config method) (kiwi.systems.tlm_system.TLMSystem.Config method) check_model_requirement() (kiwi.systems.qe_system.QESystem.Config method) (kiwi.systems.tlm_system.TLMSystem.Config method) check_nested_options() (kiwi.data.encoders.parallel_data_encoder.VocabularyConfig method) (kiwi.data.encoders.wmt_qe_data_encoder.VocabularyConfig method) checkpoint (kiwi.lib.train.TrainerConfig attribute) CheckpointsConfig (class in kiwi.lib.train) class_name (kiwi.systems.qe_system.QESystem.Config attribute) (kiwi.systems.tlm_system.TLMSystem.Config attribute) (kiwi.training.optimizers.OptimizerConfig attribute) class_weights (kiwi.lib.search.SearchOptions attribute) (kiwi.systems.outputs.quality_estimation.WordLevelConfig attribute) ClassWeightsConfig (class in kiwi.lib.search) cli() (in module kiwi.cli) collate_fn() (kiwi.data.encoders.parallel_data_encoder.ParallelDataEncoder method) (kiwi.data.encoders.wmt_qe_data_encoder.WMTQEDataEncoder method) compute() (kiwi.metrics.metrics.CorrectMetric method) (kiwi.metrics.metrics.ExpectedErrorMetric method) (kiwi.metrics.metrics.F1MultMetric method) (kiwi.metrics.metrics.MatthewsMetric method) (kiwi.metrics.metrics.Metric method) (kiwi.metrics.metrics.NLLMetric method) (kiwi.metrics.metrics.PearsonMetric method) (kiwi.metrics.metrics.PerplexityMetric method) (kiwi.metrics.metrics.RMSEMetric method) (kiwi.metrics.metrics.SpearmanMetric method)

concat_input() (kiwi.systems.encoders.bert.BertEncoder static method) (kiwi.systems.encoders.xlm.XLMEncoder static method) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder static method) CONFIG (in module kiwi.constants) Configuration (class in kiwi.lib.evaluate) (class in kiwi.lib.predict) (class in kiwi.lib.pretrain) (class in kiwi.lib.search) (class in kiwi.lib.train) configure_logging() (in module kiwi.lib.utils) configure_optimizers() (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) configure_outputs() (kiwi.runner.Runner method) configure_seed() (in module kiwi.lib.utils) confusion_matrix() (in module kiwi.metrics.functions) contextualize_b() (kiwi.systems.encoders.predictor.DualSequencesEncoder method) convolve_tensor() (in module kiwi.utils.tensors) CorrectMetric (class in kiwi.metrics.metrics) (in module kiwi.metrics) counter (kiwi.data.vocabulary.Vocabulary attribute)

D

data (kiwi.lib.predict.Configuration attribute) (kiwi.lib.train.Configuration attribute) data_processing (kiwi.systems.qe_system.QESystem.Config attribute) (kiwi.systems.tlm_system.TLMSystem.Config attribute) DataEncoders (class in kiwi.data.encoders.base) debug (kiwi.lib.train.Configuration attribute) decode_outputs() (kiwi.systems.outputs.quality_estimation.QEOutputs method) decode_sentence_outputs() (kiwi.systems.outputs.quality_estimation.QEOutputs static method) decode_word_outputs() (kiwi.systems.outputs.quality_estimation.QEOutputs method) decoder (kiwi.systems.bert.ModelConfig attribute) (kiwi.systems.nuqe.ModelConfig attribute) (kiwi.systems.predictor_estimator.ModelConfig attribute) (kiwi.systems.qe_system.ModelConfig attribute) (kiwi.systems.xlm.ModelConfig attribute) (kiwi.systems.xlmroberta.ModelConfig attribute) default_map_location() (in module kiwi.utils.io) DefaultFrozenDict (class in kiwi.utils.data_structures)

delta_average() (in module kiwi.metrics.functions) deterministic (kiwi.lib.train.TrainerConfig attribute) detokenize() (in module kiwi.data.tokenizers) dim (kiwi.modules.token_embeddings.TokenEmbeddings.Config attribute) directory (kiwi.lib.search.Configuration attribute) distribution (kiwi.lib.search.RangeConfig attribute) dropout (kiwi.lib.search.SearchOptions attribute) (kiwi.modules.token_embeddings.TokenEmbeddings.Config attribute) (kiwi.systems.decoders.estimator.EstimatorDecoder.Config attribute) (kiwi.systems.decoders.linear.LinearDecoder.Config attribute) (kiwi.systems.decoders.nuqe.NuQESourceDecoder.Config attribute) (kiwi.systems.decoders.nuqe.NuQETargetDecoder.Config attribute) (kiwi.systems.encoders.predictor.PredictorEncoder.Config attribute) (kiwi.systems.outputs.quality_estimation.QEOutputs.Config attribute) dropout_on_rnns() (kiwi.systems.decoders.estimator.EstimatorDecoder.Config method) (kiwi.systems.encoders.predictor.PredictorEncoder.Config method) DualSequencesEncoder (class in kiwi.systems.encoders.predictor)

E

early_stop_patience (kiwi.lib.train.CheckpointsConfig attribute) embeddings (kiwi.data.encoders.parallel_data_encoder.ParallelDataEncoder.Config attribute) (kiwi.data.encoders.wmt_qe_data_encoder.WMTQEDataEncoder.Config attribute) (kiwi.systems.encoders.predictor.PredictorEncoder.Config attribute) (kiwi.systems.encoders.quetch.QUETCHEncoder.Config attribute) EmbeddingsConfig (class in kiwi.data.encoders.parallel_data_encoder) (class in kiwi.data.encoders.wmt_qe_data_encoder) encode() (kiwi.data.encoders.field_encoders.AlignmentEncoder method) (kiwi.data.encoders.field_encoders.BinaryScoreEncoder method) (kiwi.data.encoders.field_encoders.ScoreEncoder method) (kiwi.data.encoders.field_encoders.TextEncoder method) encode_b() (kiwi.systems.encoders.predictor.DualSequencesEncoder method) encoder (kiwi.systems.bert.ModelConfig attribute) (kiwi.systems.nuqe.ModelConfig attribute) (kiwi.systems.predictor.ModelConfig attribute) (kiwi.systems.predictor_estimator.ModelConfig attribute) (kiwi.systems.qe_system.ModelConfig attribute) (kiwi.systems.tlm_system.ModelConfig attribute) (kiwi.systems.xlm.ModelConfig attribute) (kiwi.systems.xlmroberta.ModelConfig attribute)

encoder_learning_rate (kiwi.training.optimizers.OptimizerConfig attribute) enforce_loading() (kiwi.lib.predict.Configuration method) ensure_list() (kiwi.lib.evaluate.Configuration method) ensure_there_is_validation_data() (kiwi.data.datasets.parallel_dataset.ParallelDataset.Config method) (kiwi.data.datasets.wmt_qe_dataset.WMTQEDataset.Config method) eos_id() (kiwi.data.vocabulary.Vocabulary property) epochs (kiwi.lib.train.TrainerConfig attribute) EstimatorDecoder (class in kiwi.systems.decoders.estimator) EstimatorDecoder.Config (class in kiwi.systems.decoders.estimator) eval_sentence_level() (in module kiwi.lib.evaluate) eval_word_level() (in module kiwi.lib.evaluate) evaluate_from_configuration() (in module kiwi.lib.evaluate) ExpectedErrorMetric (class in kiwi.metrics.metrics) (in module kiwi.metrics) experiment_name (kiwi.lib.train.RunConfig attribute) extra (kiwi.utils.io.BaseConfig.Config attribute)

F

f1_product() (in module kiwi.metrics.functions) f1_scores() (in module kiwi.metrics.functions) F1MultMetric (class in kiwi.metrics.metrics) (in module kiwi.metrics) feedforward() (in module kiwi.modules.common.feedforward) file_path() (in module kiwi.assets.config) file_to_configuration() (in module kiwi.lib.utils) fine_tune (kiwi.systems.outputs.translation_language_model.TLMOutputs.Config attribute) fit_vocab() (kiwi.data.encoders.field_encoders.TextEncoder method) (kiwi.systems.encoders.bert.TransformersTextEncoder method) (kiwi.systems.encoders.xlm.XLMTextEncoder method) (kiwi.systems.encoders.xlmroberta.XLMRobertaTextEncoder method) fit_vocabularies() (kiwi.data.encoders.parallel_data_encoder.ParallelDataEncoder method) (kiwi.data.encoders.wmt_qe_data_encoder.WMTQEDataEncoder method) fix_relative_path() (kiwi.systems.encoders.bert.BertEncoder.Config method) (kiwi.systems.encoders.xlm.XLMEncoder.Config method) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder.Config method) format (kiwi.data.encoders.parallel_data_encoder.EmbeddingsConfig attribute) (kiwi.data.encoders.wmt_qe_data_encoder.EmbeddingsConfig attribute) forward() (kiwi.modules.common.attention.Attention method) (kiwi.modules.common.layer_norm.TFLayerNorm method) (kiwi.modules.common.positional_encoding.PositionalEncoding method) (kiwi.modules.common.scalar_mix.ScalarMixWithDropout method) (kiwi.modules.common.scorer.MLPScorer method) (kiwi.modules.common.scorer.Scorer method) (kiwi.modules.sentence_level_output.BinarySentenceScore method) (kiwi.modules.sentence_level_output.SentenceFromLogits method) (kiwi.modules.sentence_level_output.SentenceScoreDistribution method) (kiwi.modules.sentence_level_output.SentenceScoreRegression method) (kiwi.modules.token_embeddings.TokenEmbeddings method) (kiwi.modules.word_level_output.GapTagsOutput method) (kiwi.modules.word_level_output.WordLevelOutput method) (kiwi.systems.decoders.estimator.EstimatorDecoder method) (kiwi.systems.decoders.linear.LinearDecoder method) (kiwi.systems.decoders.nuqe.NuQEDecoder method) (kiwi.systems.decoders.nuqe.NuQESourceDecoder method) (kiwi.systems.decoders.nuqe.NuQETargetDecoder method) (kiwi.systems.encoders.bert.BertEncoder method) (kiwi.systems.encoders.predictor.DualSequencesEncoder method) (kiwi.systems.encoders.predictor.PredictorEncoder method) (kiwi.systems.encoders.quetch.QUETCHEncoder method) (kiwi.systems.encoders.xlm.XLMEncoder method) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder method) (kiwi.systems.outputs.quality_estimation.QEOutputs method) (kiwi.systems.outputs.translation_language_model.MaskedWordOutput method) (kiwi.systems.outputs.translation_language_model.TLMOutputs method) (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) (kiwi.utils.tensors.GradientMul static method)

freeze (kiwi.modules.token_embeddings.TokenEmbeddings.Config attribute) (kiwi.systems.encoders.bert.BertEncoder.Config attribute) (kiwi.systems.encoders.xlm.XLMEncoder.Config attribute) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder.Config attribute) freeze_epochs (kiwi.lib.search.SearchOptions attribute) freeze_for_number_of_steps (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder.Config attribute) from_config() (in module kiwi.training.optimizers) (kiwi.systems.qe_system.QESystem static method) (kiwi.systems.tlm_system.TLMSystem static method) from_dict() (kiwi.systems._meta_module.MetaModule class method) (kiwi.systems._meta_module.Serializable class method) (kiwi.systems.qe_system.QESystem class method) (kiwi.systems.tlm_system.TLMSystem class method) fscore() (in module kiwi.metrics.functions)

G

GAP_TAGS (in module kiwi.constants) gap_tags (kiwi.lib.evaluate.OutputConfig attribute) (kiwi.lib.search.ClassWeightsConfig attribute) gap_tags_BAD_probabilities (kiwi.runner.Predictions attribute) gap_tags_labels (kiwi.runner.Predictions attribute) gaps (kiwi.systems.outputs.quality_estimation.WordLevelConfig attribute) GapTagsOutput (class in kiwi.modules.word_level_output) gelu() (in module kiwi.modules.common.activations) generate_slug() (in module kiwi.utils.io) get_artifact_uri() (kiwi.loggers.MLFlowTrackingLogger method) get_mismatch_features() (kiwi.systems.encoders.bert.BertEncoder static method) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder static method)

get_noam_decay_schedule() (in module kiwi.training.optimizers) get_predictions_flat() (kiwi.metrics.metrics.LabeledMetric method) get_suggestion() (in module kiwi.lib.search) get_target_flat() (kiwi.metrics.metrics.LabeledMetric method) gold_files (kiwi.lib.evaluate.Configuration attribute) gpu_id (kiwi.lib.predict.Configuration attribute) GPUConfig (class in kiwi.lib.train) gpus (kiwi.lib.train.GPUConfig attribute) gradient_accumulation_steps (kiwi.lib.train.TrainerConfig attribute) gradient_max_norm (kiwi.lib.train.TrainerConfig attribute) gradient_mul (in module kiwi.utils.tensors) GradientMul (class in kiwi.utils.tensors)

H

handle_example() (in module kiwi.cli) has_rsample (kiwi.modules.common.distributions.TruncatedNormal attribute) hidden_size (kiwi.lib.search.SearchOptions attribute) (kiwi.systems.decoders.estimator.EstimatorDecoder.Config attribute) (kiwi.systems.decoders.linear.LinearDecoder.Config attribute) (kiwi.systems.encoders.bert.BertEncoder.Config attribute) (kiwi.systems.encoders.predictor.PredictorEncoder.Config attribute) (kiwi.systems.encoders.xlm.XLMEncoder.Config attribute) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder.Config attribute)

hidden_sizes (kiwi.systems.decoders.nuqe.NuQESourceDecoder.Config attribute) (kiwi.systems.decoders.nuqe.NuQETargetDecoder.Config attribute) hter (kiwi.systems.outputs.quality_estimation.SentenceLevelConfig attribute)

I

icdf() (kiwi.modules.common.distributions.TruncatedNormal method) id_to_token() (kiwi.data.vocabulary.Vocabulary method) input (kiwi.data.datasets.parallel_dataset.TestConfig attribute) (kiwi.data.datasets.parallel_dataset.TrainingConfig attribute) (kiwi.data.datasets.wmt_qe_dataset.TestConfig attribute) (kiwi.data.datasets.wmt_qe_dataset.TrainingConfig attribute) input_data_encoders() (kiwi.systems.encoders.bert.BertEncoder class method) (kiwi.systems.encoders.predictor.PredictorEncoder class method) (kiwi.systems.encoders.quetch.QUETCHEncoder class method) (kiwi.systems.encoders.xlm.XLMEncoder class method) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder class method) input_layer_norm (kiwi.modules.token_embeddings.TokenEmbeddings.Config attribute) InputConfig (class in kiwi.data.datasets.parallel_dataset) (class in kiwi.data.datasets.wmt_qe_dataset)

InputEmbeddingsConfig (class in kiwi.systems.encoders.quetch) InputEncoder (class in kiwi.data.encoders.field_encoders) InputFields (class in kiwi.data.encoders.parallel_data_encoder) (class in kiwi.data.encoders.wmt_qe_data_encoder) insert_dummy_outputs_for_empty_sentences() (kiwi.runner.Runner static method) interleave_input (kiwi.systems.encoders.bert.BertEncoder.Config attribute) (kiwi.systems.encoders.xlm.XLMEncoder.Config attribute) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder.Config attribute) interleave_input() (kiwi.systems.encoders.bert.BertEncoder static method) (kiwi.systems.encoders.xlm.XLMEncoder static method) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder static method) itos (kiwi.data.vocabulary.Vocabulary attribute)

K

keep_rare_words_with_embeddings (kiwi.data.encoders.parallel_data_encoder.VocabularyConfig attribute) (kiwi.data.encoders.wmt_qe_data_encoder.VocabularyConfig attribute) kiwi module kiwi.__main__ module kiwi.assets module kiwi.assets.config module kiwi.cli module kiwi.constants module kiwi.data module kiwi.data.batch module kiwi.data.datasets module kiwi.data.datasets.parallel_dataset module kiwi.data.datasets.wmt_qe_dataset module kiwi.data.encoders module kiwi.data.encoders.base module kiwi.data.encoders.field_encoders module kiwi.data.encoders.parallel_data_encoder module kiwi.data.encoders.wmt_qe_data_encoder module kiwi.data.tokenizers module kiwi.data.vocabulary module kiwi.lib module kiwi.lib.evaluate module kiwi.lib.predict module kiwi.lib.pretrain module kiwi.lib.search module kiwi.lib.train module kiwi.lib.utils module kiwi.loggers module kiwi.metrics module kiwi.metrics.functions module kiwi.metrics.metrics module kiwi.modules module kiwi.modules.common module kiwi.modules.common.activations module kiwi.modules.common.attention module kiwi.modules.common.distributions module kiwi.modules.common.feedforward module kiwi.modules.common.layer_norm module

kiwi.modules.common.positional_encoding module kiwi.modules.common.scalar_mix module kiwi.modules.common.scorer module kiwi.modules.sentence_level_output module kiwi.modules.token_embeddings module kiwi.modules.word_level_output module kiwi.runner module kiwi.systems module kiwi.systems._meta_module module kiwi.systems.bert module kiwi.systems.decoders module kiwi.systems.decoders.estimator module kiwi.systems.decoders.linear module kiwi.systems.decoders.nuqe module kiwi.systems.encoders module kiwi.systems.encoders.bert module kiwi.systems.encoders.predictor module kiwi.systems.encoders.quetch module kiwi.systems.encoders.xlm module kiwi.systems.encoders.xlmroberta module kiwi.systems.nuqe module kiwi.systems.outputs module kiwi.systems.outputs.quality_estimation module kiwi.systems.outputs.translation_language_model module kiwi.systems.predictor module kiwi.systems.predictor_estimator module kiwi.systems.qe_system module kiwi.systems.tlm_system module kiwi.systems.xlm module kiwi.systems.xlmroberta module kiwi.training module kiwi.training.callbacks module kiwi.training.optimizers module kiwi.utils module kiwi.utils.data_structures module kiwi.utils.io module kiwi.utils.tensors module

L

LabeledMetric (class in kiwi.metrics.metrics) LABELS (in module kiwi.constants) labels() (kiwi.systems.outputs.quality_estimation.QEOutputs method) (kiwi.systems.outputs.translation_language_model.TLMOutputs method) last_activation (kiwi.systems.outputs.quality_estimation.QEOutputs.Config attribute) learning_rate (kiwi.lib.search.SearchOptions attribute) (kiwi.training.optimizers.OptimizerConfig attribute) learning_rate_decay (kiwi.training.optimizers.OptimizerConfig attribute) learning_rate_decay_start (kiwi.training.optimizers.OptimizerConfig attribute) lengths (kiwi.data.batch.BatchedSentence attribute) lengths_match() (in module kiwi.lib.evaluate) LinearDecoder (class in kiwi.systems.decoders.linear) LinearDecoder.Config (class in kiwi.systems.decoders.linear) load (kiwi.systems.qe_system.QESystem.Config attribute) (kiwi.systems.tlm_system.TLMSystem.Config attribute) (kiwi.training.optimizers.OptimizerConfig attribute) load() (kiwi.systems._meta_module.Serializable class method) (kiwi.systems.qe_system.QESystem class method) (kiwi.systems.tlm_system.TLMSystem class method) load_encoder (kiwi.systems.qe_system.QESystem.Config attribute) load_from_checkpoint() (kiwi.systems.qe_system.QESystem class method) (kiwi.systems.tlm_system.TLMSystem class method) load_state_dict() (kiwi.systems.encoders.bert.BertEncoder method) (kiwi.systems.encoders.xlm.XLMEncoder method) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder method) load_study (kiwi.lib.search.Configuration attribute) load_system() (in module kiwi.lib.predict) load_torch_file() (in module kiwi.utils.io) load_vocabs (kiwi.systems.qe_system.QESystem.Config attribute) (kiwi.systems.tlm_system.TLMSystem.Config attribute) load_vocabularies() (kiwi.data.encoders.parallel_data_encoder.ParallelDataEncoder method) (kiwi.data.encoders.wmt_qe_data_encoder.WMTQEDataEncoder method) log_artifact() (kiwi.loggers.MLFlowTrackingLogger method) log_artifacts() (kiwi.loggers.MLFlowTrackingLogger method) log_hyperparams() (kiwi.loggers.MLFlowTrackingLogger method) log_interval (kiwi.lib.train.TrainerConfig attribute) log_metrics() (kiwi.loggers.MLFlowTrackingLogger method) log_model() (kiwi.loggers.MLFlowTrackingLogger method) log_param() (kiwi.loggers.MLFlowTrackingLogger method) log_prob() (kiwi.modules.common.distributions.TruncatedNormal method) log_save_interval (kiwi.lib.train.TrainerConfig attribute)

log_tag() (kiwi.loggers.MLFlowTrackingLogger method) log_tags() (kiwi.loggers.MLFlowTrackingLogger method) logger (in module kiwi.data.datasets.parallel_dataset) (in module kiwi.data.datasets.wmt_qe_dataset) (in module kiwi.data.encoders.field_encoders) (in module kiwi.data.encoders.parallel_data_encoder) (in module kiwi.data.encoders.wmt_qe_data_encoder) (in module kiwi.data.vocabulary) (in module kiwi.lib.evaluate) (in module kiwi.lib.predict) (in module kiwi.lib.pretrain) (in module kiwi.lib.search) (in module kiwi.lib.train) (in module kiwi.loggers) (in module kiwi.runner) (in module kiwi.systems._meta_module) (in module kiwi.systems.bert) (in module kiwi.systems.decoders.estimator) (in module kiwi.systems.encoders.bert) (in module kiwi.systems.encoders.predictor) (in module kiwi.systems.encoders.quetch) (in module kiwi.systems.encoders.xlm) (in module kiwi.systems.encoders.xlmroberta) (in module kiwi.systems.nuqe) (in module kiwi.systems.outputs.quality_estimation) (in module kiwi.systems.predictor) (in module kiwi.systems.predictor_estimator) (in module kiwi.systems.qe_system) (in module kiwi.systems.tlm_system) (in module kiwi.systems.xlm) (in module kiwi.systems.xlmroberta) (in module kiwi.training.callbacks) (in module kiwi.training.optimizers) (in module kiwi.utils.io) LOSS (in module kiwi.constants) loss() (kiwi.systems.outputs.quality_estimation.QEOutputs method) (kiwi.systems.outputs.translation_language_model.TLMOutputs method) (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) loss_fn() (kiwi.modules.sentence_level_output.SentenceFromLogits method) lower (kiwi.lib.search.RangeConfig attribute)

M

main() (in module kiwi.__main__) main_metric (kiwi.lib.train.TrainerConfig attribute) main_metric() (kiwi.lib.search.Objective property) (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) make_classes_loss_weights() (in module kiwi.utils.tensors) make_mergeable_tensors() (in module kiwi.utils.tensors) make_predictions() (in module kiwi.lib.predict) map_name_to_class() (kiwi.systems.qe_system.QESystem.Config method) (kiwi.systems.tlm_system.TLMSystem.Config method) MaskedWordOutput (class in kiwi.systems.outputs.translation_language_model) matthews_correlation_coefficient() (in module kiwi.metrics.functions) MatthewsMetric (class in kiwi.metrics.metrics) (in module kiwi.metrics) max_position_embeddings (kiwi.modules.token_embeddings.TokenEmbeddings.Config attribute) max_size (kiwi.data.encoders.parallel_data_encoder.VocabularyConfig attribute) (kiwi.data.encoders.wmt_qe_data_encoder.VocabularyConfig attribute) max_size() (kiwi.data.vocabulary.Vocabulary method) mean() (kiwi.modules.common.distributions.TruncatedNormal property) mean_absolute_error() (in module kiwi.metrics.functions) mean_squared_error() (in module kiwi.metrics.functions) MetaModule (class in kiwi.systems._meta_module) MetaModule.Config (class in kiwi.systems._meta_module) Metric (class in kiwi.metrics.metrics) metrics() (kiwi.systems.outputs.quality_estimation.QEOutputs property) (kiwi.systems.outputs.translation_language_model.TLMOutputs property) metrics_end() (kiwi.systems.outputs.quality_estimation.QEOutputs method) (kiwi.systems.outputs.translation_language_model.TLMOutputs method) (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) metrics_step() (kiwi.systems.outputs.quality_estimation.QEOutputs method) (kiwi.systems.outputs.translation_language_model.TLMOutputs method) (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) MetricsReport (class in kiwi.lib.evaluate) min_frequency (kiwi.data.encoders.parallel_data_encoder.VocabularyConfig attribute) (kiwi.data.encoders.wmt_qe_data_encoder.VocabularyConfig attribute) mlflow_always_log_artifacts (kiwi.lib.train.RunConfig attribute) mlflow_tracking_uri (kiwi.lib.train.RunConfig attribute) MLFlowTrackingLogger (class in kiwi.loggers) MLPScorer (class in kiwi.modules.common.scorer) model (kiwi.lib.train.TrainRunInfo attribute) (kiwi.systems.bert.Bert.Config attribute) (kiwi.systems.nuqe.NuQE.Config attribute) (kiwi.systems.predictor.Predictor.Config attribute) (kiwi.systems.predictor_estimator.PredictorEstimator.Config attribute) (kiwi.systems.qe_system.QESystem.Config attribute) (kiwi.systems.tlm_system.TLMSystem.Config attribute) (kiwi.systems.xlm.XLM.Config attribute) (kiwi.systems.xlmroberta.XLMRoberta.Config attribute) model_name (kiwi.systems.encoders.bert.BertEncoder.Config attribute) (kiwi.systems.encoders.xlm.XLMEncoder.Config attribute) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder.Config attribute) ModelConfig (class in kiwi.systems.bert) (class in kiwi.systems.nuqe) (class in kiwi.systems.predictor) (class in kiwi.systems.predictor_estimator) (class in kiwi.systems.qe_system) (class in kiwi.systems.tlm_system) (class in kiwi.systems.xlm) (class in kiwi.systems.xlmroberta) module kiwi kiwi.__main__ kiwi.assets kiwi.assets.config kiwi.cli kiwi.constants kiwi.data kiwi.data.batch kiwi.data.datasets kiwi.data.datasets.parallel_dataset kiwi.data.datasets.wmt_qe_dataset kiwi.data.encoders kiwi.data.encoders.base kiwi.data.encoders.field_encoders kiwi.data.encoders.parallel_data_encoder kiwi.data.encoders.wmt_qe_data_encoder kiwi.data.tokenizers kiwi.data.vocabulary kiwi.lib kiwi.lib.evaluate kiwi.lib.predict kiwi.lib.pretrain kiwi.lib.search kiwi.lib.train kiwi.lib.utils kiwi.loggers kiwi.metrics kiwi.metrics.functions kiwi.metrics.metrics kiwi.modules kiwi.modules.common kiwi.modules.common.activations kiwi.modules.common.attention kiwi.modules.common.distributions kiwi.modules.common.feedforward kiwi.modules.common.layer_norm kiwi.modules.common.positional_encoding kiwi.modules.common.scalar_mix kiwi.modules.common.scorer kiwi.modules.sentence_level_output kiwi.modules.token_embeddings kiwi.modules.word_level_output kiwi.runner kiwi.systems kiwi.systems._meta_module kiwi.systems.bert kiwi.systems.decoders kiwi.systems.decoders.estimator kiwi.systems.decoders.linear kiwi.systems.decoders.nuqe kiwi.systems.encoders kiwi.systems.encoders.bert kiwi.systems.encoders.predictor kiwi.systems.encoders.quetch kiwi.systems.encoders.xlm kiwi.systems.encoders.xlmroberta kiwi.systems.nuqe kiwi.systems.outputs kiwi.systems.outputs.quality_estimation kiwi.systems.outputs.translation_language_model kiwi.systems.predictor kiwi.systems.predictor_estimator kiwi.systems.qe_system kiwi.systems.tlm_system kiwi.systems.xlm kiwi.systems.xlmroberta kiwi.training kiwi.training.callbacks kiwi.training.optimizers kiwi.utils kiwi.utils.data_structures kiwi.utils.io kiwi.utils.tensors

module (in module kiwi.systems) MultiFieldBatch (class in kiwi.data.batch)

N

n_layers_output (kiwi.systems.outputs.quality_estimation.QEOutputs.Config attribute) name() (kiwi.metrics.metrics.Metric property) net_length() (kiwi.data.vocabulary.Vocabulary method) NLLMetric (class in kiwi.metrics.metrics) no_implementation() (kiwi.systems.encoders.bert.BertEncoder.Config method) (kiwi.systems.encoders.predictor.PredictorEncoder.Config method) (kiwi.systems.encoders.xlm.XLMEncoder.Config method) normalize_metric_key() (in module kiwi.loggers) normalize_prediction_files() (in module kiwi.lib.evaluate) num_data_workers (kiwi.systems.qe_system.QESystem.Config attribute) (kiwi.systems.tlm_system.TLMSystem.Config attribute) num_embeddings() (kiwi.modules.token_embeddings.TokenEmbeddings property) num_models_to_keep (kiwi.lib.search.Configuration attribute)

num_parameters() (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) num_tokens() (kiwi.metrics.metrics.Metric method) num_train_lines() (kiwi.lib.search.Objective property) num_trials (kiwi.lib.search.Configuration attribute) number_of_tokens (kiwi.data.batch.BatchedSentence attribute) NuQE (class in kiwi.systems.nuqe) NuQE.Config (class in kiwi.systems.nuqe) NuQEDecoder (class in kiwi.systems.decoders.nuqe) NuQEDecoder.Config (class in kiwi.systems.decoders.nuqe) NuQESourceDecoder (class in kiwi.systems.decoders.nuqe) NuQESourceDecoder.Config (class in kiwi.systems.decoders.nuqe) NuQETargetDecoder (class in kiwi.systems.decoders.nuqe) NuQETargetDecoder.Config (class in kiwi.systems.decoders.nuqe)

O

Objective (class in kiwi.lib.search) OK (in module kiwi.constants) on_load_checkpoint() (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) on_save_checkpoint() (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) on_train_begin() (kiwi.training.callbacks.BestMetricsInfo method) on_train_end() (kiwi.training.callbacks.BestMetricsInfo method) on_validation_end() (kiwi.training.callbacks.BestMetricsInfo method) optimizer (kiwi.systems.qe_system.QESystem.Config attribute) (kiwi.systems.tlm_system.TLMSystem.Config attribute) optimizer_class() (in module kiwi.training.optimizers) optimizer_name() (in module kiwi.training.optimizers) OptimizerConfig (class in kiwi.training.optimizers)

OPTIMIZERS_MAPPING (in module kiwi.training.optimizers) options (kiwi.lib.search.Configuration attribute) out_embeddings_dim (kiwi.systems.encoders.predictor.PredictorEncoder.Config attribute) output (kiwi.data.datasets.wmt_qe_dataset.TrainingConfig attribute) output_dir (kiwi.lib.predict.RunConfig attribute) (kiwi.lib.train.RunConfig attribute) OutputConfig (class in kiwi.data.datasets.wmt_qe_dataset) (class in kiwi.lib.evaluate) outputs (kiwi.systems.bert.ModelConfig attribute) (kiwi.systems.nuqe.ModelConfig attribute) (kiwi.systems.predictor_estimator.ModelConfig attribute) (kiwi.systems.qe_system.ModelConfig attribute) (kiwi.systems.xlm.ModelConfig attribute) (kiwi.systems.xlmroberta.ModelConfig attribute)

P

package_dir (in module kiwi.systems) PAD (in module kiwi.constants) pad_id() (kiwi.data.vocabulary.Vocabulary property) pad_zeros_around_timesteps() (in module kiwi.utils.tensors) padding_index() (kiwi.data.encoders.field_encoders.TextEncoder property) ParallelDataEncoder (class in kiwi.data.encoders.parallel_data_encoder) ParallelDataEncoder.Config (class in kiwi.data.encoders.parallel_data_encoder) ParallelDataset (class in kiwi.data.datasets.parallel_dataset) ParallelDataset.Config (class in kiwi.data.datasets.parallel_dataset) parse_base_config() (kiwi.lib.search.Configuration method) partition_function() (kiwi.modules.common.distributions.TruncatedNormal method) patience (kiwi.lib.search.SearchOptions attribute) PE (in module kiwi.constants) PE_LOGITS (in module kiwi.constants) PearsonMetric (class in kiwi.metrics.metrics) (in module kiwi.metrics) PerplexityMetric (class in kiwi.metrics.metrics) (in module kiwi.metrics) pieces_to_tokens() (in module kiwi.utils.tensors) pin_memory() (kiwi.data.batch.BatchedSentence method) (kiwi.data.batch.MultiFieldBatch method) pooling (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder.Config attribute) PositionalEncoding (class in kiwi.modules.common.positional_encoding) post_edit (kiwi.data.datasets.wmt_qe_dataset.InputConfig attribute) (kiwi.data.encoders.wmt_qe_data_encoder.EmbeddingsConfig attribute)

precision (kiwi.lib.train.GPUConfig attribute) precision() (in module kiwi.metrics.functions) precision_recall_fscore_support() (in module kiwi.metrics.functions) predict() (kiwi.runner.Runner method) (kiwi.systems.qe_system.QESystem method) predict_from_configuration() (in module kiwi.lib.predict) predict_on_data_partition (kiwi.lib.predict.RunConfig attribute) predicted_dir (kiwi.lib.evaluate.Configuration attribute) predicted_files (kiwi.lib.evaluate.Configuration attribute) Predictions (class in kiwi.runner) Predictor (class in kiwi.systems.predictor) Predictor.Config (class in kiwi.systems.predictor) PredictorEncoder (class in kiwi.systems.encoders.predictor) PredictorEncoder.Config (class in kiwi.systems.encoders.predictor) PredictorEstimator (class in kiwi.systems.predictor_estimator) PredictorEstimator.Config (class in kiwi.systems.predictor_estimator) prepare_data() (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) prepare_dataloader() (kiwi.systems.qe_system.QESystem method) pretrain_from_configuration() (in module kiwi.lib.pretrain) pretrain_from_file() (in module kiwi.lib.pretrain) print_scores_table() (kiwi.lib.evaluate.MetricsReport method) prune_models() (kiwi.lib.search.Objective method)

Q

QEOutputs (class in kiwi.systems.outputs.quality_estimation) QEOutputs.Config (class in kiwi.systems.outputs.quality_estimation) QESystem (class in kiwi.systems.qe_system) QESystem.Config (class in kiwi.systems.qe_system) QUETCHEncoder (class in kiwi.systems.encoders.quetch)

QUETCHEncoder.Config (class in kiwi.systems.encoders.quetch) quiet (kiwi.lib.evaluate.Configuration attribute) (kiwi.lib.predict.Configuration attribute) (kiwi.lib.search.Configuration attribute) (kiwi.lib.train.Configuration attribute)

R

RangeConfig (class in kiwi.lib.search) read_file() (in module kiwi.data.datasets.parallel_dataset) (in module kiwi.data.datasets.wmt_qe_dataset) (in module kiwi.utils.io) read_sentence_scores_file() (in module kiwi.lib.evaluate) recall() (in module kiwi.metrics.functions) register_subclass() (kiwi.systems._meta_module.Serializable class method) remove_empty_sentences() (kiwi.runner.Runner static method) replace_token() (in module kiwi.utils.tensors) report_lengths_mismatch() (in module kiwi.lib.evaluate) resume (kiwi.lib.train.TrainerConfig attribute) retrieve_gold_standard() (in module kiwi.lib.evaluate) retrieve_subclass() (kiwi.systems._meta_module.Serializable class method) retrieve_tokens_mask() (in module kiwi.utils.tensors) RMSEMetric (class in kiwi.metrics.metrics) (in module kiwi.metrics)

rnn_layers (kiwi.systems.decoders.estimator.EstimatorDecoder.Config attribute) (kiwi.systems.encoders.predictor.PredictorEncoder.Config attribute) run (kiwi.lib.predict.Configuration attribute) (kiwi.lib.train.Configuration attribute) run() (in module kiwi.lib.evaluate) (in module kiwi.lib.predict) (in module kiwi.lib.search) (in module kiwi.lib.train) (kiwi.runner.Runner method) run_id (kiwi.lib.predict.RunConfig attribute) (kiwi.lib.train.RunConfig attribute) RunConfig (class in kiwi.lib.predict) (class in kiwi.lib.train) Runner (class in kiwi.runner)

S

save() (kiwi.systems._meta_module.Serializable method) save_config_to_file() (in module kiwi.lib.utils) save_file() (in module kiwi.utils.io) save_predicted_probabilities() (in module kiwi.utils.io) save_top_k (kiwi.lib.train.CheckpointsConfig attribute) scalar_mix_dropout (kiwi.systems.encoders.bert.BertEncoder.Config attribute) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder.Config attribute) scalar_mix_layer_norm (kiwi.systems.encoders.bert.BertEncoder.Config attribute) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder.Config attribute) ScalarMixWithDropout (class in kiwi.modules.common.scalar_mix) scale() (kiwi.modules.common.distributions.TruncatedNormal property) (kiwi.modules.common.scorer.Scorer method) scale_embeddings (kiwi.modules.token_embeddings.TokenEmbeddings.Config attribute) ScoreEncoder (class in kiwi.data.encoders.field_encoders) Scorer (class in kiwi.modules.common.scorer) scores_for_class() (in module kiwi.metrics.functions) search_from_configuration() (in module kiwi.lib.search) search_from_file() (in module kiwi.lib.search) search_hter (kiwi.lib.search.SearchOptions attribute) search_method (kiwi.lib.search.SearchOptions attribute) search_mlp (kiwi.lib.search.SearchOptions attribute) search_name (kiwi.lib.search.Configuration attribute) search_word_level (kiwi.lib.search.SearchOptions attribute) SearchOptions (class in kiwi.lib.search) seed (kiwi.lib.predict.RunConfig attribute) (kiwi.lib.search.Configuration attribute) (kiwi.lib.train.RunConfig attribute) select_positions() (in module kiwi.utils.tensors) sentence_level (kiwi.systems.outputs.quality_estimation.QEOutputs.Config attribute) sentence_level_scores() (in module kiwi.lib.evaluate) sentence_loss_weight (kiwi.lib.search.SearchOptions attribute) (kiwi.systems.outputs.quality_estimation.QEOutputs.Config attribute) sentence_losses() (kiwi.systems.outputs.quality_estimation.QEOutputs method) SENTENCE_SCORES (in module kiwi.constants) sentence_scores (kiwi.data.datasets.wmt_qe_dataset.OutputConfig attribute) (kiwi.lib.evaluate.MetricsReport attribute) SentenceFromLogits (class in kiwi.modules.sentence_level_output) SentenceLevelConfig (class in kiwi.systems.outputs.quality_estimation) SentenceMetric (class in kiwi.metrics.metrics) sentences_hter (kiwi.runner.Predictions attribute) SentenceScoreDistribution (class in kiwi.modules.sentence_level_output) SentenceScoreRegression (class in kiwi.modules.sentence_level_output) sequence_mask() (in module kiwi.utils.tensors) Serializable (class in kiwi.systems._meta_module) set_config_options() (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) setup_amp_level() (kiwi.lib.train.GPUConfig method) setup_gpu() (kiwi.lib.predict.Configuration method) setup_gpu_id() (kiwi.lib.predict.Configuration method) setup_gpu_ids() (kiwi.lib.train.GPUConfig method) setup_output_directory() (in module kiwi.lib.utils) setup_run() (in module kiwi.lib.predict) (in module kiwi.lib.search) (in module kiwi.lib.train) share_embeddings (kiwi.systems.encoders.predictor.PredictorEncoder.Config attribute) share_input_fields_encoders (kiwi.data.encoders.parallel_data_encoder.ParallelDataEncoder.Config attribute) (kiwi.data.encoders.wmt_qe_data_encoder.WMTQEDataEncoder.Config attribute) size() (kiwi.modules.token_embeddings.TokenEmbeddings method) (kiwi.systems.decoders.estimator.EstimatorDecoder method) (kiwi.systems.decoders.linear.LinearDecoder method) (kiwi.systems.decoders.nuqe.NuQEDecoder method) (kiwi.systems.decoders.nuqe.NuQESourceDecoder method) (kiwi.systems.decoders.nuqe.NuQETargetDecoder method) (kiwi.systems.encoders.bert.BertEncoder method) (kiwi.systems.encoders.predictor.PredictorEncoder method) (kiwi.systems.encoders.quetch.QUETCHEncoder method) (kiwi.systems.encoders.xlm.XLMEncoder method) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder method)

sort_by_second_value() (in module kiwi.lib.utils) sort_key() (kiwi.data.datasets.parallel_dataset.ParallelDataset method) (kiwi.data.datasets.wmt_qe_dataset.WMTQEDataset method) SOURCE (in module kiwi.constants) source (kiwi.data.datasets.parallel_dataset.InputConfig attribute) (kiwi.data.datasets.wmt_qe_dataset.InputConfig attribute) (kiwi.data.encoders.parallel_data_encoder.EmbeddingsConfig attribute) (kiwi.data.encoders.parallel_data_encoder.InputFields attribute) (kiwi.data.encoders.wmt_qe_data_encoder.EmbeddingsConfig attribute) (kiwi.data.encoders.wmt_qe_data_encoder.InputFields attribute) (kiwi.systems.decoders.nuqe.NuQEDecoder.Config attribute) (kiwi.systems.encoders.quetch.InputEmbeddingsConfig attribute) (kiwi.systems.outputs.quality_estimation.WordLevelConfig attribute) source_language (kiwi.systems.encoders.xlm.XLMEncoder.Config attribute) SOURCE_LOGITS (in module kiwi.constants) SOURCE_POS (in module kiwi.constants) source_pos (kiwi.data.datasets.wmt_qe_dataset.InputConfig attribute) (kiwi.data.encoders.wmt_qe_data_encoder.EmbeddingsConfig attribute) (kiwi.systems.encoders.quetch.InputEmbeddingsConfig attribute) SOURCE_TAGS (in module kiwi.constants) source_tags (kiwi.data.datasets.wmt_qe_dataset.OutputConfig attribute) (kiwi.lib.search.ClassWeightsConfig attribute) source_tags_BAD_probabilities (kiwi.runner.Predictions attribute) source_tags_labels (kiwi.runner.Predictions attribute) sparse_embeddings (kiwi.modules.token_embeddings.TokenEmbeddings.Config attribute) SpearmanMetric (class in kiwi.metrics.metrics) (in module kiwi.metrics) split (kiwi.data.datasets.parallel_dataset.ParallelDataset.Config attribute) (kiwi.data.datasets.wmt_qe_dataset.WMTQEDataset.Config attribute) split_outputs() (kiwi.systems.encoders.bert.BertEncoder static method) (kiwi.systems.encoders.xlm.XLMEncoder static method) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder static method) split_wmt18_tags() (in module kiwi.lib.evaluate) START (in module kiwi.constants) STATE_DICT (in module kiwi.constants) step (kiwi.lib.search.RangeConfig attribute) step() (kiwi.metrics.metrics.CorrectMetric method) (kiwi.metrics.metrics.ExpectedErrorMetric method) (kiwi.metrics.metrics.LabeledMetric method) (kiwi.metrics.metrics.Metric method) (kiwi.metrics.metrics.NLLMetric method) (kiwi.metrics.metrics.PerplexityMetric method) (kiwi.metrics.metrics.SentenceMetric method) stoi (kiwi.data.vocabulary.Vocabulary attribute) STOP (in module kiwi.constants) strict_masks (kiwi.data.batch.BatchedSentence attribute) subclasses (kiwi.systems._meta_module.Serializable attribute) (kiwi.systems.qe_system.QESystem attribute) (kiwi.systems.tlm_system.TLMSystem attribute) suggest_train_config() (kiwi.lib.search.Objective method) support (kiwi.modules.common.distributions.TruncatedNormal attribute) swish() (in module kiwi.modules.common.activations) system (kiwi.lib.predict.Configuration attribute) (kiwi.lib.pretrain.Configuration attribute) (kiwi.lib.train.Configuration attribute)

T

T (in module kiwi.data.encoders.parallel_data_encoder) (in module kiwi.data.encoders.wmt_qe_data_encoder) tag_metrics() (in module kiwi.systems.outputs.quality_estimation) TagEncoder (class in kiwi.data.encoders.field_encoders) TARGET (in module kiwi.constants) target (kiwi.data.datasets.parallel_dataset.InputConfig attribute) (kiwi.data.datasets.wmt_qe_dataset.InputConfig attribute) (kiwi.data.encoders.parallel_data_encoder.EmbeddingsConfig attribute) (kiwi.data.encoders.parallel_data_encoder.InputFields attribute) (kiwi.data.encoders.wmt_qe_data_encoder.EmbeddingsConfig attribute) (kiwi.data.encoders.wmt_qe_data_encoder.InputFields attribute) (kiwi.systems.decoders.nuqe.NuQEDecoder.Config attribute) (kiwi.systems.encoders.quetch.InputEmbeddingsConfig attribute) (kiwi.systems.outputs.quality_estimation.WordLevelConfig attribute) target_gaps_to_gaps() (in module kiwi.utils.io) target_gaps_to_target() (in module kiwi.utils.io) target_language (kiwi.systems.encoders.xlm.XLMEncoder.Config attribute) TARGET_LOGITS (in module kiwi.constants) TARGET_POS (in module kiwi.constants) target_pos (kiwi.data.datasets.wmt_qe_dataset.InputConfig attribute) (kiwi.data.encoders.wmt_qe_data_encoder.EmbeddingsConfig attribute) (kiwi.systems.encoders.quetch.InputEmbeddingsConfig attribute) TARGET_SENTENCE (in module kiwi.constants) TARGET_TAGS (in module kiwi.constants) target_tags (kiwi.data.datasets.wmt_qe_dataset.OutputConfig attribute) (kiwi.lib.search.ClassWeightsConfig attribute) target_tags_BAD_probabilities (kiwi.runner.Predictions attribute) target_tags_labels (kiwi.runner.Predictions attribute) TARGETGAPS_TAGS (in module kiwi.constants) targetgaps_tags (kiwi.lib.evaluate.OutputConfig attribute) tensor (kiwi.data.batch.BatchedSentence attribute) tensors_to() (in module kiwi.data.batch) test (kiwi.data.datasets.parallel_dataset.ParallelDataset.Config attribute) (kiwi.data.datasets.wmt_qe_dataset.WMTQEDataset.Config attribute) (kiwi.systems.qe_system.BatchSizeConfig attribute) test_dataloader() (kiwi.systems.qe_system.QESystem method) test_epoch_end() (kiwi.systems.qe_system.QESystem method) test_step() (kiwi.systems.qe_system.QESystem method) TestConfig (class in kiwi.data.datasets.parallel_dataset) (class in kiwi.data.datasets.wmt_qe_dataset) TextEncoder (class in kiwi.data.encoders.field_encoders) TFLayerNorm (class in kiwi.modules.common.layer_norm)

tlm_outputs (kiwi.systems.bert.ModelConfig attribute) (kiwi.systems.predictor.ModelConfig attribute) (kiwi.systems.predictor_estimator.ModelConfig attribute) (kiwi.systems.qe_system.ModelConfig attribute) (kiwi.systems.tlm_system.ModelConfig attribute) (kiwi.systems.xlm.ModelConfig attribute) (kiwi.systems.xlmroberta.ModelConfig attribute) TLMOutputs (class in kiwi.systems.outputs.translation_language_model) TLMOutputs.Config (class in kiwi.systems.outputs.translation_language_model) TLMSystem (class in kiwi.systems.tlm_system) TLMSystem.Config (class in kiwi.systems.tlm_system) to() (kiwi.data.batch.BatchedSentence method) (kiwi.data.batch.MultiFieldBatch method) to_dict() (kiwi.systems._meta_module.MetaModule method) (kiwi.systems._meta_module.Serializable class method) (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) to_numeric_binary_labels() (in module kiwi.lib.evaluate) to_numeric_values() (in module kiwi.lib.evaluate) token_to_id() (kiwi.data.vocabulary.Vocabulary method) TokenEmbeddings (class in kiwi.modules.token_embeddings) TokenEmbeddings.Config (class in kiwi.modules.token_embeddings) tokenize() (in module kiwi.data.tokenizers) tracking_uri() (kiwi.loggers.MLFlowTrackingLogger property) train (kiwi.data.datasets.parallel_dataset.ParallelDataset.Config attribute) (kiwi.data.datasets.wmt_qe_dataset.WMTQEDataset.Config attribute) (kiwi.systems.qe_system.BatchSizeConfig attribute) (kiwi.systems.tlm_system.BatchSizeConfig attribute) train_dataloader() (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) train_from_configuration() (in module kiwi.lib.train) train_from_file() (in module kiwi.lib.train) trainer (kiwi.lib.train.Configuration attribute) TrainerConfig (class in kiwi.lib.train) training_epoch_end() (kiwi.systems.qe_system.QESystem method) training_step() (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) training_steps (kiwi.training.optimizers.OptimizerConfig attribute) TrainingConfig (class in kiwi.data.datasets.parallel_dataset) (class in kiwi.data.datasets.wmt_qe_dataset) TrainRunInfo (class in kiwi.lib.train) TransformersTextEncoder (class in kiwi.systems.encoders.bert) TruncatedNormal (class in kiwi.modules.common.distributions)

U

UNALIGNED (in module kiwi.constants) UNK (in module kiwi.constants) unmask() (in module kiwi.utils.tensors) unsqueeze_as() (in module kiwi.utils.tensors) updates_per_epochs() (kiwi.lib.search.Objective property) upper (kiwi.lib.search.RangeConfig attribute) use_distribution (kiwi.systems.outputs.quality_estimation.SentenceLevelConfig attribute) use_gpu (kiwi.lib.predict.Configuration attribute) use_mismatch_features (kiwi.systems.encoders.bert.BertEncoder.Config attribute) (kiwi.systems.encoders.predictor.PredictorEncoder.Config attribute) (kiwi.systems.encoders.xlm.XLMEncoder.Config attribute)

use_mlflow (kiwi.lib.train.RunConfig attribute) use_mlp (kiwi.systems.decoders.estimator.EstimatorDecoder.Config attribute) (kiwi.systems.encoders.bert.BertEncoder.Config attribute) (kiwi.systems.encoders.xlm.XLMEncoder.Config attribute) (kiwi.systems.encoders.xlmroberta.XLMRobertaEncoder.Config attribute) use_position_embeddings (kiwi.modules.token_embeddings.TokenEmbeddings.Config attribute) use_predictor_features (kiwi.systems.encoders.bert.BertEncoder.Config attribute) (kiwi.systems.encoders.xlm.XLMEncoder.Config attribute) use_v0_buggy_strategy (kiwi.systems.decoders.estimator.EstimatorDecoder.Config attribute) (kiwi.systems.encoders.predictor.PredictorEncoder.Config attribute)

V

v0_start_stop (kiwi.systems.encoders.predictor.PredictorEncoder.Config attribute) val_dataloader() (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) valid (kiwi.data.datasets.parallel_dataset.ParallelDataset.Config attribute) (kiwi.data.datasets.wmt_qe_dataset.WMTQEDataset.Config attribute) (kiwi.systems.qe_system.BatchSizeConfig attribute) (kiwi.systems.tlm_system.BatchSizeConfig attribute) validate_metric_value() (in module kiwi.loggers) validation_epoch_end() (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) validation_step() (kiwi.systems.qe_system.QESystem method) (kiwi.systems.tlm_system.TLMSystem method) validation_steps (kiwi.lib.search.SearchOptions attribute) (kiwi.lib.train.CheckpointsConfig attribute) variance() (kiwi.modules.common.distributions.TruncatedNormal property)

verbose (kiwi.lib.evaluate.Configuration attribute) (kiwi.lib.predict.Configuration attribute) (kiwi.lib.search.Configuration attribute) (kiwi.lib.train.Configuration attribute) VOCAB (in module kiwi.constants) vocab (kiwi.data.encoders.parallel_data_encoder.ParallelDataEncoder.Config attribute) (kiwi.data.encoders.wmt_qe_data_encoder.WMTQEDataEncoder.Config attribute) vocabularies() (kiwi.data.encoders.base.DataEncoders property) (kiwi.data.encoders.parallel_data_encoder.ParallelDataEncoder property) (kiwi.data.encoders.wmt_qe_data_encoder.WMTQEDataEncoder property) vocabularies_from_dict() (kiwi.data.encoders.parallel_data_encoder.ParallelDataEncoder method) (kiwi.data.encoders.wmt_qe_data_encoder.WMTQEDataEncoder method) Vocabulary (class in kiwi.data.vocabulary) vocabulary() (kiwi.data.encoders.field_encoders.TextEncoder property) VocabularyConfig (class in kiwi.data.encoders.parallel_data_encoder) (class in kiwi.data.encoders.wmt_qe_data_encoder)

W

warmup_steps (kiwi.lib.search.SearchOptions attribute) (kiwi.training.optimizers.OptimizerConfig attribute) warn_missing_feature() (kiwi.data.encoders.parallel_data_encoder.ParallelDataEncoder.Config method) (kiwi.data.encoders.wmt_qe_data_encoder.WMTQEDataEncoder.Config method) window_size (kiwi.systems.encoders.quetch.QUETCHEncoder.Config attribute) WMTQEDataEncoder (class in kiwi.data.encoders.wmt_qe_data_encoder) WMTQEDataEncoder.Config (class in kiwi.data.encoders.wmt_qe_data_encoder) WMTQEDataset (class in kiwi.data.datasets.wmt_qe_dataset)

WMTQEDataset.Config (class in kiwi.data.datasets.wmt_qe_dataset) word_level (kiwi.systems.outputs.quality_estimation.QEOutputs.Config attribute) word_level_scores() (in module kiwi.lib.evaluate) word_losses() (kiwi.systems.outputs.quality_estimation.QEOutputs method) word_scores (kiwi.lib.evaluate.MetricsReport attribute) WordLevelConfig (class in kiwi.systems.outputs.quality_estimation) WordLevelOutput (class in kiwi.modules.word_level_output) wrap_predictions() (kiwi.runner.Runner static method)

X

XLM (class in kiwi.systems.xlm) XLM.Config (class in kiwi.systems.xlm) XLMEncoder (class in kiwi.systems.encoders.xlm) XLMEncoder.Config (class in kiwi.systems.encoders.xlm) XLMRoberta (class in kiwi.systems.xlmroberta)

XLMRoberta.Config (class in kiwi.systems.xlmroberta) XLMRobertaEncoder (class in kiwi.systems.encoders.xlmroberta) XLMRobertaEncoder.Config (class in kiwi.systems.encoders.xlmroberta) XLMRobertaTextEncoder (class in kiwi.systems.encoders.xlmroberta) XLMTextEncoder (class in kiwi.systems.encoders.xlm)