Enum rust_bert::pipelines::question_answering::QuestionAnsweringOption

pub enum QuestionAnsweringOption {
    Bert(BertForQuestionAnswering),
    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)

Tuple Fields

0: BertForQuestionAnswering

Bert for Question Answering

DistilBert(DistilBertForQuestionAnswering)

Tuple Fields

0: DistilBertForQuestionAnswering

DistilBert for Question Answering

MobileBert(MobileBertForQuestionAnswering)

Tuple Fields

0: MobileBertForQuestionAnswering

MobileBert for Question Answering

Roberta(RobertaForQuestionAnswering)

Tuple Fields

0: RobertaForQuestionAnswering

Roberta for Question Answering

XLMRoberta(RobertaForQuestionAnswering)

Tuple Fields

0: RobertaForQuestionAnswering

XLMRoberta for Question Answering

Albert(AlbertForQuestionAnswering)

Tuple Fields

0: AlbertForQuestionAnswering

Albert for Question Answering

XLNet(XLNetForQuestionAnswering)

Tuple Fields

0: XLNetForQuestionAnswering

XLNet for Question Answering

Reformer(ReformerForQuestionAnswering)

Tuple Fields

0: ReformerForQuestionAnswering

Reformer for Question Answering

Longformer(LongformerForQuestionAnswering)

Tuple Fields

0: LongformerForQuestionAnswering

Longformer for Question Answering

FNet(FNetForQuestionAnswering)

Tuple Fields

0: FNetForQuestionAnswering

FNet for Question Answering

Implementations

impl QuestionAnsweringOption

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

Instantiate a new question answering 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)

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>,
    train: bool
) -> (Tensor, Tensor)

Interface method to forward_t() of the particular models.