Enum rust_bert::pipelines::question_answering::QuestionAnsweringOption

pub enum QuestionAnsweringOption {
    Bert(BertForQuestionAnswering),
    Deberta(DebertaForQuestionAnswering),
    DebertaV2(DebertaV2ForQuestionAnswering),
    DistilBert(DistilBertForQuestionAnswering),
    MobileBert(MobileBertForQuestionAnswering),
    Roberta(RobertaForQuestionAnswering),
    XLMRoberta(RobertaForQuestionAnswering),
    Albert(AlbertForQuestionAnswering),
    XLNet(XLNetForQuestionAnswering),
    Reformer(ReformerForQuestionAnswering),
    Longformer(LongformerForQuestionAnswering),
    FNet(FNetForQuestionAnswering),
}

Abstraction that holds one particular question answering model, for any of the supported models

Variants

Bert(BertForQuestionAnswering)

Bert for Question Answering

Deberta(DebertaForQuestionAnswering)

DeBERTa for Question Answering

DebertaV2(DebertaV2ForQuestionAnswering)

DeBERTa V2 for Question Answering

DistilBert(DistilBertForQuestionAnswering)

DistilBert for Question Answering

MobileBert(MobileBertForQuestionAnswering)

MobileBert for Question Answering

Roberta(RobertaForQuestionAnswering)

Roberta for Question Answering

XLMRoberta(RobertaForQuestionAnswering)

XLMRoberta for Question Answering

Albert(AlbertForQuestionAnswering)

Albert for Question Answering

XLNet(XLNetForQuestionAnswering)

XLNet for Question Answering

Reformer(ReformerForQuestionAnswering)

Reformer for Question Answering

Longformer(LongformerForQuestionAnswering)

Longformer for Question Answering

FNet(FNetForQuestionAnswering)

FNet for Question Answering

Implementations

impl QuestionAnsweringOption

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

Instantiate a new question answering model of the supplied type.

Arguments

• QuestionAnsweringConfig - Question answering 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 SequenceClassificationOption

pub fn forward_t( &self, input_ids: Option<&Tensor>, mask: Option<&Tensor>, input_embeds: Option<&Tensor>, _token_type_ids: Option<&Tensor>, train: bool ) -> (Tensor, Tensor)

Interface method to forward_t() of the particular models.