lm_polygraph.stat_calculators.statistic_extraction module

class lm_polygraph.stat_calculators.statistic_extraction.TrainingStatisticExtractionCalculator(train_dataset=None, background_train_dataset=None)[source]

Bases: StatCalculator

static meta_info() → Tuple[List[str], List[str]][source]

Returns the statistics and dependencies for the calculator.