Enum rust_bert::pipelines::token_classification::TokenClassificationOption

pub enum TokenClassificationOption {
    Bert(BertForTokenClassification),
    Deberta(DebertaForTokenClassification),
    DebertaV2(DebertaV2ForTokenClassification),
    DistilBert(DistilBertForTokenClassification),
    MobileBert(MobileBertForTokenClassification),
    Roberta(RobertaForTokenClassification),
    XLMRoberta(RobertaForTokenClassification),
    Electra(ElectraForTokenClassification),
    Albert(AlbertForTokenClassification),
    XLNet(XLNetForTokenClassification),
    Longformer(LongformerForTokenClassification),
    FNet(FNetForTokenClassification),
}

Abstraction that holds one particular token sequence classifier model, for any of the supported models

Variants

Bert(BertForTokenClassification)

Bert for Token Classification

Deberta(DebertaForTokenClassification)

DeBERTa for Token Classification

DebertaV2(DebertaV2ForTokenClassification)

DeBERTa V2 for Token Classification

DistilBert(DistilBertForTokenClassification)

DistilBert for Token Classification

MobileBert(MobileBertForTokenClassification)

MobileBert for Token Classification

Roberta(RobertaForTokenClassification)

Roberta for Token Classification

XLMRoberta(RobertaForTokenClassification)

XLM Roberta for Token Classification

Electra(ElectraForTokenClassification)

Electra for Token Classification

Albert(AlbertForTokenClassification)

Albert for Token Classification

XLNet(XLNetForTokenClassification)

XLNet for Token Classification

Longformer(LongformerForTokenClassification)

Longformer for Token Classification

FNet(FNetForTokenClassification)

FNet for Token Classification

Implementations

impl TokenClassificationOption

pub fn new(config: &TokenClassificationConfig) -> Result<Self, RustBertError>

Instantiate a new sequence classification model of the supplied type.

Arguments

• TokenClassificationConfig - Token classification pipeline configuration. The type of model created will be inferred from the ModelResources (Torch or ONNX) and ModelType (Architecture for Torch models) variants provided and

pub fn model_type(&self) -> ModelType

Returns the ModelType for this TokenClassificationOption