[−][src]Enum rust_bert::pipelines::sequence_classification::SequenceClassificationOption
Variants
Bert for Sequence Classification
DistilBert(DistilBertModelClassifier)
DistilBert for Sequence Classification
Roberta(RobertaForSequenceClassification)
Roberta for Sequence Classification
XLMRoberta(RobertaForSequenceClassification)
XLMRoberta for Sequence Classification
Albert(AlbertForSequenceClassification)
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]
P: Borrow<Path<'p>>,
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 formodel_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]
&self,
input_ids: Option<Tensor>,
mask: Option<Tensor>,
token_type_ids: Option<Tensor>,
position_ids: Option<Tensor>,
input_embeds: Option<Tensor>,
train: bool
) -> Tensor
Interface method to forward_t() of the particular models.
Auto Trait Implementations
impl !RefUnwindSafe for SequenceClassificationOption
impl !Send for SequenceClassificationOption
impl !Sync for SequenceClassificationOption
impl Unpin for SequenceClassificationOption
impl !UnwindSafe for SequenceClassificationOption
Blanket Implementations
impl<T> Any for T where
T: 'static + ?Sized,
[src]
T: 'static + ?Sized,
impl<T> Borrow<T> for T where
T: ?Sized,
[src]
T: ?Sized,
impl<T> BorrowMut<T> for T where
T: ?Sized,
[src]
T: ?Sized,
fn borrow_mut(&mut self) -> &mut T
[src]
impl<T> From<T> for T
[src]
impl<T, U> Into<U> for T where
U: From<T>,
[src]
U: From<T>,
impl<T, U> TryFrom<U> for T where
U: Into<T>,
[src]
U: Into<T>,
type Error = Infallible
The type returned in the event of a conversion error.
fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>
[src]
impl<T, U> TryInto<U> for T where
U: TryFrom<T>,
[src]
U: TryFrom<T>,
type Error = <U as TryFrom<T>>::Error
The type returned in the event of a conversion error.
fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>
[src]
impl<V, T> VZip<V> for T where
V: MultiLane<T>,
V: MultiLane<T>,