API reference

The API reference is auto-generated from the source code (docstrings and type hints) using mkdocstrings.

By area:

• Models — GradiendModel, ParamMappedGradiendModel, ModelWithGradiend
• Data — TextFilterConfig, TextPredictionDataCreator, TextPreprocessConfig
• Training — TrainingArguments, Trainer, TextPredictionTrainer, pruning configs (PrePruneConfig, PostPruneConfig)
• Evaluation — Evaluator, EncoderEvaluator, DecoderEvaluator, Visualizer
• Visualization — plot functions (plot_training_convergence, plot_topk_overlap_heatmap, …)
• Comparison — compute_similarity_matrix, compute_gradiend_feature_cross_encoding_matrix, …

Use the navigation sidebar to open each area’s index and the listed classes.