pub struct SequenceClassificationConfig {
pub model_type: ModelType,
pub model_resource: ModelResource,
pub config_resource: Box<dyn ResourceProvider + Send>,
pub vocab_resource: Box<dyn ResourceProvider + Send>,
pub merges_resource: Option<Box<dyn ResourceProvider + Send>>,
pub lower_case: bool,
pub strip_accents: Option<bool>,
pub add_prefix_space: Option<bool>,
pub device: Device,
pub kind: Option<Kind>,
}
Expand description
§Configuration for SequenceClassificationModel
Contains information regarding the model to load and device to place the model on.
Fields§
§model_type: ModelType
Model type
model_resource: ModelResource
Model weights resource (default: pretrained BERT model on CoNLL)
config_resource: Box<dyn ResourceProvider + Send>
Config resource (default: pretrained BERT model on CoNLL)
vocab_resource: Box<dyn ResourceProvider + Send>
Vocab resource (default: pretrained BERT model on CoNLL)
merges_resource: Option<Box<dyn ResourceProvider + Send>>
Merges resource (default: None)
lower_case: bool
Automatically lower case all input upon tokenization (assumes a lower-cased model)
strip_accents: Option<bool>
Flag indicating if the tokenizer should strip accents (normalization). Only used for BERT / ALBERT models
add_prefix_space: Option<bool>
Flag indicating if the tokenizer should add a white space before each tokenized input (needed for some Roberta models)
device: Device
Device to place the model on (default: CUDA/GPU when available)
kind: Option<Kind>
Model weights precision. If not provided, will default to full precision on CPU, or the loaded weights precision otherwise
Implementations§
Source§impl SequenceClassificationConfig
impl SequenceClassificationConfig
Sourcepub fn new<RC, RV>(
model_type: ModelType,
model_resource: ModelResource,
config_resource: RC,
vocab_resource: RV,
merges_resource: Option<RV>,
lower_case: bool,
strip_accents: impl Into<Option<bool>>,
add_prefix_space: impl Into<Option<bool>>,
) -> SequenceClassificationConfig
pub fn new<RC, RV>( model_type: ModelType, model_resource: ModelResource, config_resource: RC, vocab_resource: RV, merges_resource: Option<RV>, lower_case: bool, strip_accents: impl Into<Option<bool>>, add_prefix_space: impl Into<Option<bool>>, ) -> SequenceClassificationConfig
Instantiate a new sequence classification configuration of the supplied type.
§Arguments
model_type
-ModelType
indicating the model type to load (must match with the actual data to be loaded!)- model - The
ResourceProvider
pointing to the model to load (e.g. model.ot) - config - The
ResourceProvider
pointing to the model configuration to load (e.g. config.json) - vocab - The
ResourceProvider
pointing to the tokenizer’s vocabulary to load (e.g. vocab.txt/vocab.json) - vocab - An optional
ResourceProvider
pointing to the tokenizer’s merge file to load (e.g. merges.txt), needed only for Roberta. - lower_case - A
bool
indicating whether the tokenizer should lower case all input (in case of a lower-cased model)
Trait Implementations§
Source§impl Default for SequenceClassificationConfig
impl Default for SequenceClassificationConfig
Source§fn default() -> SequenceClassificationConfig
fn default() -> SequenceClassificationConfig
Provides a defaultSST-2 sentiment analysis model (English)
Auto Trait Implementations§
impl Freeze for SequenceClassificationConfig
impl !RefUnwindSafe for SequenceClassificationConfig
impl Send for SequenceClassificationConfig
impl Sync for SequenceClassificationConfig
impl Unpin for SequenceClassificationConfig
impl !UnwindSafe for SequenceClassificationConfig
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
self
into a Left
variant of Either<Self, Self>
if into_left
is true
.
Converts self
into a Right
variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
self
into a Left
variant of Either<Self, Self>
if into_left(&self)
returns true
.
Converts self
into a Right
variant of Either<Self, Self>
otherwise. Read more