textfier.tasks¶
Pre-defined tasks provide an easier framework when loading pre-trained models. Hence, this package serves as a wrapper around pre-trained models loading from Huggingface’s Transformers.
Task-related classes and helpers with pre-defined tasks.
- class textfier.tasks.CausalLanguageModelingTask(model: str, **kwargs)¶
Bases:
textfier.core.Task
CausalLanguageModelingTask implements a pre-trained task used to handle causal language modeling.
- __init__(self, model: str, **kwargs)¶
Initialization method.
- Parameters
model – Identifier of the pre-trained model to be loaded.
- _build(self, model: str)¶
Builds up the pre-trained model according to the desired task.
- Parameters
model – Identifier of the pre-trained model to be built.
- class textfier.tasks.MaskedLanguageModelingTask(model: str, **kwargs)¶
Bases:
textfier.core.Task
MaskedLanguageModelingTask implements a pre-trained task used to handle masked language modeling.
- __init__(self, model: str, **kwargs)¶
Initialization method.
- Parameters
model – Identifier of the pre-trained model to be loaded.
- _build(self, model: str)¶
Builds up the pre-trained model according to the desired task.
- Parameters
model – Identifier of the pre-trained model to be built.
- class textfier.tasks.NamedEntityRecognitionTask(model: str, **kwargs)¶
Bases:
textfier.core.Task
NamedEntityRecognitionTask implements a pre-trained task used to handle named entity recognition.
- __init__(self, model: str, **kwargs)¶
Initialization method.
- Parameters
model – Identifier of the pre-trained model to be loaded.
- _build(self, model: str)¶
Builds up the pre-trained model according to the desired task.
- Parameters
model – Identifier of the pre-trained model to be built.
- class textfier.tasks.QuestionAnsweringTask(model: str, **kwargs)¶
Bases:
textfier.core.Task
QuestionAnsweringTask implements a pre-trained task used to handle question answering.
- __init__(self, model: str, **kwargs)¶
Initialization method.
- Parameters
model – Identifier of the pre-trained model to be loaded.
- _build(self, model: str)¶
Builds up the pre-trained model according to the desired task.
- Parameters
model – Identifier of the pre-trained model to be built.
- class textfier.tasks.Seq2SeqTask(model: str, **kwargs)¶
Bases:
textfier.core.Task
Seq2SeqTask implements a pre-trained task used to handle sequence-to-sequence tasks (summarization and translation).
- __init__(self, model: str, **kwargs)¶
Initialization method.
- Parameters
model – Identifier of the pre-trained model to be loaded.
- _build(self, model: str)¶
Builds up the pre-trained model according to the desired task.
- Parameters
model – Identifier of the pre-trained model to be built.
- class textfier.tasks.SequenceClassificationTask(model: str, **kwargs)¶
Bases:
textfier.core.Task
SequenceClassificationTask implements a pre-trained task used to handle text classification.
- __init__(self, model: str, **kwargs)¶
Initialization method.
- Parameters
model – Identifier of the pre-trained model to be loaded.
- _build(self, model: str)¶
Builds up the pre-trained model according to the desired task.
- Parameters
model – Identifier of the pre-trained model to be built.