ComprehendItPredictor
Predictor for knowledgator/comprehend_it models
This predictor is specifically build for knowledgator/comprehend_it models. For more details about models, see:
About use cases and more, see:
Subclass of TransformersPipeline.
Module: implementation.predictors
Methods and properties
Main methods and properties
__init__
Arguments:
cfg (Optional[TransformersPipelineConfig], optional): Configuration for predictor. If value equals to None, default ComprehendItPredictorConfig configuration will be used. Defaults to None.
input_class (Type[Input], optional): Class for input validation. Defaults to ComprehendItPredictorInput.
output_class (Type[Output], optional): Class for output validation. Defaults to ComprehendItPredictorOutput.
name (Optional[str], optional): Name for identification. If equals to None, class name will be used. Defaults to None.
ComprehendItPredictorConfig
Prebuild configuration that describes default parameters for knowledgator/comprehend_it models pipeline. Subclass of TransformersPipelineConfig. It contains all parameters described in TransformersPipelineConfig but overwrites some default values that described here.
__init__
Arguments:
task (Optional[str], optional): Pipeline task. Defaults to "zero-shot-classification".
model (Optional[Union[str, PreTrainedModel, TFPreTrainedModel]], optional): Model that will be used. Defaults to "knowledgator/comprehend_it-base".
kwargs (Optional[Dict[str, Any]], optional): Extra parameters. Defaults to
...
For other parameters, see TransformersPipelineConfig.
ComprehendItPredictorInput
Subclass of IOModel.
__init__
Arguments:
sequences (str): Text to classify.
candidate_labels (List[str]): Classification labels to use.
ComprehendItPredictorOutput
Subclass of IOModel.
__init__
Arguments:
sequence (str): Text used for classification.
labels (List[str]): Classified labels.
scores (List[float]): Classification scores.
Last updated
