langtest.modelhandler.custom_modelhandler.PretrainedModelForCrowsPairs#
- class PretrainedModelForCrowsPairs(model: Any)#
Bases:
PretrainedCustomModel
,ModelAPI
A custom model handler for pretrained models that are used for crows pairs.
Methods
__init__
(model)load_model
(path)Load the model.
predict
(text, *args, **kwargs)Perform predictions on input text.
predict_raw
(text, *args, **kwargs)Attributes
model_registry
- classmethod load_model(path: Any) Any #
Load the model.
- abstract predict(text: str, *args, **kwargs)#
Perform predictions on input text.