Struct tokenizers::models::wordlevel::WordLevelTrainer

#[non_exhaustive]
pub struct WordLevelTrainer {
    pub min_frequency: u64,
    pub vocab_size: usize,
    pub show_progress: bool,
    pub special_tokens: Vec<AddedToken>,
    /* private fields */
}

Fields (Non-exhaustive)

Non-exhaustive structs could have additional fields added in future. Therefore, non-exhaustive structs cannot be constructed in external crates using the traditional Struct { .. } syntax; cannot be matched against without a wildcard ..; and struct update syntax will not work.

min_frequency: u64

The minimum frequency a word must have to be part of the vocabulary

vocab_size: usize

The target vocabulary size

show_progress: bool

Whether to show progress while training

special_tokens: Vec<AddedToken>

A list of special tokens that the model should know of

Implementations

impl WordLevelTrainer

pub fn builder() -> WordLevelTrainerBuilder

Trait Implementations

impl Clone for WordLevelTrainer

fn clone(&self) -> WordLevelTrainer

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for WordLevelTrainer

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

Formats the value using the given formatter.

impl Default for WordLevelTrainer

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for WordLevelTrainer

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl From<WordLevelTrainer> for TrainerWrapper

fn from(from: WordLevelTrainer) -> Self

Converts to this type from the input type.

impl Serialize for WordLevelTrainer

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Trainer for WordLevelTrainer

fn train(&self, model: &mut WordLevel) -> Result<Vec<AddedToken>>

Train a WordLevel model

fn should_show_progress(&self) -> bool

Whether we should show progress

type Model = WordLevel

fn feed<I, S, F>(&mut self, iterator: I, process: F) -> Result<()>

where I: Iterator<Item = S> + Send, S: AsRef<str> + Send, F: Fn(&str) -> Result<Vec<String>> + Sync,

Process an iterator of sequences, calling process for each of them in order to pre-process the said sequence as relevant.