Enum rust_bert::pipelines::token_classification::TokenClassificationOption[−][src]

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

Expand description

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

Variants

`Bert(BertForTokenClassification)`

Tuple Fields

0: BertForTokenClassification

Bert for Token Classification

`DistilBert(DistilBertForTokenClassification)`

Tuple Fields

0: DistilBertForTokenClassification

DistilBert for Token Classification

`MobileBert(MobileBertForTokenClassification)`

Tuple Fields

0: MobileBertForTokenClassification

MobileBert for Token Classification

`Roberta(RobertaForTokenClassification)`

Tuple Fields

0: RobertaForTokenClassification

Roberta for Token Classification

`XLMRoberta(RobertaForTokenClassification)`

Tuple Fields

0: RobertaForTokenClassification

XLM Roberta for Token Classification

`Electra(ElectraForTokenClassification)`

Tuple Fields

0: ElectraForTokenClassification

Electra for Token Classification

`Albert(AlbertForTokenClassification)`

Tuple Fields

0: AlbertForTokenClassification

Albert for Token Classification

`XLNet(XLNetForTokenClassification)`

Tuple Fields

0: XLNetForTokenClassification

XLNet for Token Classification

`Longformer(LongformerForTokenClassification)`

Tuple Fields

0: LongformerForTokenClassification

Longformer for Token Classification

`FNet(FNetForTokenClassification)`

Tuple Fields

0: FNetForTokenClassification

FNet for Token Classification

Implementations

[src]

impl TokenClassificationOption

[src]

pub fn new<'p, P>(
 model_type: ModelType,
 p: P,
 config: &ConfigOption
) -> Result<Self, RustBertError> where
 P: Borrow<Path<'p>>,

Instantiate a new token sequence classification model of the supplied type.

Arguments

model_type - ModelType indicating the model type to load (must match with the actual data to be loaded)
p - tch::nn::Path path to the model file to load (e.g. model.ot)
config - A configuration (the model type of the configuration must be compatible with the value for model_type)

[src]