Enum onnxruntime::download::language::machine_comprehension::MachineComprehension

pub enum MachineComprehension {
    BiDAF,
    BERTSquad,
    RoBERTa(RoBERTa),
    GPT2(GPT2),
}

Machine Comprehension

This subset of natural language processing models that answer questions about a given context paragraph.

Source: https://github.com/onnx/models#machine_comprehension

Variants

BiDAF

Answers a query about a given context paragraph.

This model is a neural network for answering a query about a given context paragraph.

Source: https://github.com/onnx/models/tree/master/text/machine_comprehension/bidirectional_attention_flow

Variant downloaded: ONNX Version 1.4 with Opset Version 9.

BERTSquad

Answers questions based on the context of the given input paragraph.

Source: https://github.com/onnx/models/tree/master/text/machine_comprehension/bert-squad

Variant downloaded: ONNX Version 1.5 with Opset Version 10.

RoBERTa(RoBERTa)

Large transformer-based model that predicts sentiment based on given input text.

Transformer-based language model for text generation.

Source: https://github.com/onnx/models/tree/master/text/machine_comprehension/roberta

GPT2(GPT2)

Large transformer-based language model that given a sequence of words within some text, predicts the next word.

Source: https://github.com/onnx/models/tree/master/text/machine_comprehension/gpt-2

Trait Implementations

impl Clone for MachineComprehension

fn clone(&self) -> MachineComprehension

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for MachineComprehension

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl From<MachineComprehension> for AvailableOnnxModel

fn from(model: MachineComprehension) -> Self

Performs the conversion.