TransformersTextClassification

Task for text classification

Subclass of Task.

Module: implementation.tasks

Default predictor

This task uses TransformersModel by default with this configuration:

TransformersPipeline(
    TransformersPipelineConfig(
        task="text-classification", 
        model="ProsusAI/finbert",
        kwargs={"top_k": None}
    ),
    input_class=TransformersBasicInput,
    output_class=TransformersBasicOutput
)

See:

• TransformersPipeline

• TransformersPipelineConfig

• TransformersBasicInput

• TransformersBasicOutput

Methods and properties

Main methods and properties

---

__init__

Arguments:

• predictor (Predictor[Any, Any], optional): Predictor that will be used in task. If equals to None, default predictor will be used. Defaults to None.
• preprocess (Optional[Component], optional): Component executed before predictor. Defaults to None.
• postprocess (Optional[Component], optional): Component executed after predictor. Defaults to None.
• input_class (Type[Input], optional): Class for input validation. Defaults to TransformersBasicInput
• output_class (Type[Output], optional): Class for output validation. Defaults to TransformersBasicOutput
• name (Optional[str], optional): Name for identification. If equals to None, class name will be used. Defaults to None.

---

---