lm_polygraph.stat_calculators.sample_visual module

class lm_polygraph.stat_calculators.sample_visual.OutputWrapper[source]

Bases: object

decoder_hidden_states = None

encoder_hidden_states = None

hidden_states = None

class lm_polygraph.stat_calculators.sample_visual.SamplingGenerationVisualCalculator(samples_n: int = 10)[source]

Bases: StatCalculator

For Whitebox model (lm_polygraph.VisualWhiteboxModel), at input texts batch calculates: * sampled texts * tokens of the sampled texts * probabilities of the sampled tokens generation

static meta_info() → Tuple[List[str], List[str]][source]: Returns the statistics and dependencies for the calculator.