langtest.modelhandler.custom_modelhandler.PretrainedModelForFactualityTest#
- class PretrainedModelForFactualityTest(model: Any)#
Bases:
PretrainedCustomModel
,ModelAPI
A class representing a pretrained model for factuality test.
Inherits from PretrainedCustomModel and ModelAPI.
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.