[][src]Enum rust_bert::pipelines::sequence_classification::SequenceClassificationOption

Variants

Bert for Sequence Classification

DistilBert for Sequence Classification

Roberta for Sequence Classification

XLMRoberta for Sequence Classification

Albert for Sequence Classification

Implementations

impl SequenceClassificationOption[src]

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

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

pub fn model_type(&self) -> ModelType[src]

Returns the ModelType for this SequenceClassificationOption

pub fn forward_t(
    &self,
    input_ids: Option<Tensor>,
    mask: Option<Tensor>,
    token_type_ids: Option<Tensor>,
    position_ids: Option<Tensor>,
    input_embeds: Option<Tensor>,
    train: bool
) -> Tensor
[src]

Interface method to forward_t() of the particular models.

Auto Trait Implementations

Blanket Implementations

impl<T> Any for T where
    T: 'static + ?Sized
[src]

impl<T> Borrow<T> for T where
    T: ?Sized
[src]

impl<T> BorrowMut<T> for T where
    T: ?Sized
[src]

impl<T> From<T> for T[src]

impl<T, U> Into<U> for T where
    U: From<T>, 
[src]

impl<T, U> TryFrom<U> for T where
    U: Into<T>, 
[src]

type Error = Infallible

The type returned in the event of a conversion error.

impl<T, U> TryInto<U> for T where
    U: TryFrom<T>, 
[src]

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

impl<V, T> VZip<V> for T where
    V: MultiLane<T>,