lm_polygraph.utils.ensemble_utils package

Submodules

• lm_polygraph.utils.ensemble_utils.dropout module
  • ConsistentDropout
    • ConsistentDropout.forward()
    • ConsistentDropout.forward_share_across_tokens()
    • ConsistentDropout.identity()
  • functional_dropout()
  • functional_dropout_share()
  • replace_dropout()
  • replace_with_identity()
• lm_polygraph.utils.ensemble_utils.ensemble_beam module
  • BeamSearchEncoderDecoderOutput
    • BeamSearchEncoderDecoderOutput.beam_indices
    • BeamSearchEncoderDecoderOutput.cross_attentions
    • BeamSearchEncoderDecoderOutput.decoder_attentions
    • BeamSearchEncoderDecoderOutput.decoder_hidden_states
    • BeamSearchEncoderDecoderOutput.encoder_attentions
    • BeamSearchEncoderDecoderOutput.encoder_hidden_states
    • BeamSearchEncoderDecoderOutput.ep_uncertainties
    • BeamSearchEncoderDecoderOutput.models_beam_next_token_logits
    • BeamSearchEncoderDecoderOutput.models_scores
    • BeamSearchEncoderDecoderOutput.pe_uncertainties
    • BeamSearchEncoderDecoderOutput.scores
    • BeamSearchEncoderDecoderOutput.sequences
    • BeamSearchEncoderDecoderOutput.sequences_scores
  • EnsembleBeamSearchMixin
    • EnsembleBeamSearchMixin.beam_search()
• lm_polygraph.utils.ensemble_utils.ensemble_generator module
  • EnsembleGenerationMixin
    • EnsembleGenerationMixin.add_ensemble_models()
    • EnsembleGenerationMixin.base_seed
    • EnsembleGenerationMixin.calculate_entropy_based_measures()
    • EnsembleGenerationMixin.ensembling_mode
    • EnsembleGenerationMixin.mc
    • EnsembleGenerationMixin.mc_models_num
    • EnsembleGenerationMixin.mc_seeds
    • EnsembleGenerationMixin.models
    • EnsembleGenerationMixin.models_beam_logits_iter
    • EnsembleGenerationMixin.tokenizer
• lm_polygraph.utils.ensemble_utils.ensemble_greedy module
  • EnsembleGreedyMixin
    • EnsembleGreedyMixin.greedy_search()
  • GreedySearchEncoderDecoderOutput
    • GreedySearchEncoderDecoderOutput.cross_attentions
    • GreedySearchEncoderDecoderOutput.decoder_attentions
    • GreedySearchEncoderDecoderOutput.decoder_hidden_states
    • GreedySearchEncoderDecoderOutput.encoder_attentions
    • GreedySearchEncoderDecoderOutput.encoder_hidden_states
    • GreedySearchEncoderDecoderOutput.ep_uncertainties
    • GreedySearchEncoderDecoderOutput.models_hypo_next_token_logits
    • GreedySearchEncoderDecoderOutput.models_scores
    • GreedySearchEncoderDecoderOutput.pe_uncertainties
    • GreedySearchEncoderDecoderOutput.scores
    • GreedySearchEncoderDecoderOutput.sequences
    • GreedySearchEncoderDecoderOutput.sequences_scores
• lm_polygraph.utils.ensemble_utils.ensemble_sample module
  • EnsembleSampleMixin
    • EnsembleSampleMixin.sample()
  • SampleEncoderDecoderOutput
    • SampleEncoderDecoderOutput.cross_attentions
    • SampleEncoderDecoderOutput.decoder_attentions
    • SampleEncoderDecoderOutput.decoder_hidden_states
    • SampleEncoderDecoderOutput.encoder_attentions
    • SampleEncoderDecoderOutput.encoder_hidden_states
    • SampleEncoderDecoderOutput.ep_uncertainties
    • SampleEncoderDecoderOutput.models_beam_next_token_logits
    • SampleEncoderDecoderOutput.models_scores
    • SampleEncoderDecoderOutput.pe_uncertainties
    • SampleEncoderDecoderOutput.scores
    • SampleEncoderDecoderOutput.sequences