Enum RoBERTa

pub enum RoBERTa {
    RoBERTaBase,
    RoBERTaSequenceClassification,
}

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

Variants

RoBERTaBase

Variant with input is a sequence of words as a string. Example: “Text to encode: Hello, World”

Variant downloaded: ONNX Version 1.6 with Opset Version 11.

RoBERTaSequenceClassification

Variant with input is a sequence of words as a string including sentiment. Example: “This film is so good”

Variant downloaded: ONNX Version 1.6 with Opset Version 9.

Trait Implementations

impl Clone for RoBERTa

fn clone(&self) -> RoBERTa

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for RoBERTa

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

Formats the value using the given formatter.

impl From<RoBERTa> for AvailableOnnxModel

fn from(model: RoBERTa) -> Self

Converts to this type from the input type.