langtest.modelhandler.transformers_modelhandler#

Classes

PretrainedModelForClinical(model, **kwargs)

A class representing a pretrained model for clinical.

PretrainedModelForCrowsPairs(model)

A class representing a pretrained model for Crows-Pairs detection.

PretrainedModelForDisinformation(model, **kwargs)

A class representing a pretrained model for disinformation. Inherits: PretrainedModelForQA: The base class for pretrained models.

PretrainedModelForFactuality(model, **kwargs)

A class representing a pretrained model for factuality detection.

PretrainedModelForIdeology(model, **kwargs)

Transformers pretrained model for ideology.

PretrainedModelForLegal(model, **kwargs)

A class representing a pretrained model for legal.

PretrainedModelForNER(model, **kwargs)

Transformers pretrained model for NER tasks

PretrainedModelForQA(model, **kwargs)

A wrapper class for using a Hugging Face pretrained model for Question Answering tasks.

PretrainedModelForSecurity(model, **kwargs)

Transformers pretrained model for security.

PretrainedModelForSensitivity(model, *args, ...)

A class for handling a pretrained model for sensitivity testing.

PretrainedModelForStereoSet(model)

A class representing a pretrained model for StereoSet detection.

PretrainedModelForSummarization(model, **kwargs)

Transformers pretrained model for summarization tasks

PretrainedModelForSycophancy(model, **kwargs)

A class representing a pretrained model for Sycophancy.

PretrainedModelForTextClassification(model)

Transformers pretrained model for text classification tasks

PretrainedModelForToxicity(model, **kwargs)

Transformers pretrained model for toxicity.

PretrainedModelForTranslation(model)

Transformers pretrained model for translation tasks

PretrainedModelForWinoBias(model, *args, ...)

A class representing a pretrained model for wino-bias detection.