Struct rust_tokenizers::preprocessing::vocab::gpt2_vocab::Gpt2Vocab

pub struct Gpt2Vocab {
    pub values: HashMap<String, i64>,
    pub indices: HashMap<i64, String>,
    pub unknown_value: &'static str,
    pub special_values: HashMap<String, i64>,
    pub special_indices: HashMap<i64, String>,
}

Fields

values: HashMap<String, i64>

indices: HashMap<i64, String>

unknown_value: &'static str

special_values: HashMap<String, i64>

special_indices: HashMap<i64, String>

Implementations

impl Gpt2Vocab

pub fn bos_value() -> &'static str

pub fn eos_value() -> &'static str

Trait Implementations

impl Tokenizer<Gpt2Vocab> for Gpt2Tokenizer

fn vocab(&self) -> &Gpt2Vocab

fn tokenize_to_tokens(&self, initial_token: TokenRef) -> Vec<Token>

Tokenize a text, returns a vector of tokens (contains offset information and more)

fn convert_tokens_to_string(&self, tokens: Vec<String>) -> String

fn tokenize(&self, text: &str) -> Vec<String>

Tokenize a string, returns a vector of tokens as strings. Use tokenize_with_offsets or tokenize_to_tokens if you also want offset information.

fn tokenize_with_offsets(
    &self,
    text: &str
) -> (Vec<String>, Vec<Option<Offset>>, Vec<Vec<OffsetSize>>, Vec<Mask>)

Tokenize a string, return offset information

fn tokenize_list(&self, text_list: Vec<&str>) -> Vec<Vec<String>>

Tokenize a vector of strings, where each corresponds to for example a sentence, returns a vector of vectors of strings. Use tokenize_list_with_offsets if you also want offset information.

fn tokenize_list_with_offsets(
    &self,
    text_list: Vec<&str>
) -> Vec<(Vec<String>, Vec<Option<Offset>>, Vec<Vec<OffsetSize>>, Vec<Mask>)>

Tokenize a vector of strings, where each corresponds to for example a sentence, returns a vector of pairs consists of a vector of tokens and a list of offset information.

fn convert_tokens_to_ids(&self, tokens: &Vec<String>) -> Vec<i64>

fn encode(
    &self,
    text_1: &str,
    text_2: Option<&str>,
    max_len: usize,
    truncation_strategy: &TruncationStrategy,
    stride: usize
) -> TokenizedInput

fn encode_list(
    &self,
    text_list: Vec<&str>,
    max_len: usize,
    truncation_strategy: &TruncationStrategy,
    stride: usize
) -> Vec<TokenizedInput>

fn encode_pair_list(
    &self,
    text_list: Vec<(&str, &str)>,
    max_len: usize,
    truncation_strategy: &TruncationStrategy,
    stride: usize
) -> Vec<TokenizedInput>

fn decode_to_vec(
    &self,
    token_ids: Vec<i64>,
    skip_special_tokens: bool
) -> Vec<String>

fn decode(
    &self,
    token_ids: Vec<i64>,
    skip_special_tokens: bool,
    clean_up_tokenization_spaces: bool
) -> String

Converts a sequence of ids (integer) into astring, using the tokenizer and vocabulary with options to remove special tokens and clean up tokenization spaces. Args: * token_ids: list of tokenized input ids. Can be obtained using the encode or encode_plus methods. * skip_special_tokens: if set to True, will replace special tokens. * clean_up_tokenization_spaces: if set to True, will clean up the tokenization spaces.

fn clean_up_tokenization(&self, input_string: String) -> String

fn decode_list(
    &self,
    token_ids_list: Vec<Vec<i64>>,
    skip_special_tokens: bool,
    clean_up_tokenization_spaces: bool
) -> Vec<String>

fn build_input_with_special_tokens(
    &self,
    tokens_1: Vec<i64>,
    tokens_2: Option<Vec<i64>>,
    offsets_1: Vec<Option<Offset>>,
    offsets_2: Option<Vec<Option<Offset>>>,
    original_offsets_1: Vec<Vec<OffsetSize>>,
    original_offsets_2: Option<Vec<Vec<OffsetSize>>>,
    mask: Vec<Mask>,
    mask_2: Option<Vec<Mask>>
) -> (Vec<i64>, Vec<i8>, Vec<i8>, Vec<Option<Offset>>, Vec<Vec<OffsetSize>>, Vec<Mask>)

Build model inputs from a sequence or a pair of sequence for sequence classification tasks by concatenating and adding special tokens. A RoBERTa sequence has the following format: single sequence: X pair of sequences: A B

impl Vocab for Gpt2Vocab

fn unknown_value() -> &'static str

Associative function returning the unknown value

fn get_unknown_value(&self) -> &'static str

Returns the unknown value on an instance

fn values(&self) -> &HashMap<String, i64>

Return the map of token strings to IDs

fn indices(&self) -> &HashMap<i64, String>

Return the map of token IDs to strings

fn special_values(&self) -> &HashMap<String, i64>

Return the map of token strings to IDs

fn special_indices(&self) -> &HashMap<i64, String>

Return the map of token IDs to strings for special values

fn from_file(path: &str) -> Gpt2Vocab

Read a vocabulary file from file

fn token_to_id(&self, token: &str) -> i64

fn id_to_token(&self, id: &i64) -> String

fn read_vocab_file(path: &str) -> HashMap<String, i64>

Read a Bert-style vocab.txt file (single column, one token per line)

fn _token_to_id(
    &self,
    token: &str,
    values: &HashMap<String, i64>,
    special_values: &HashMap<String, i64>,
    unknown_value: &str
) -> Result<i64, Box<dyn Error>>

fn _id_to_token(
    &self,
    id: &i64,
    indices: &HashMap<i64, String>,
    special_indices: &HashMap<i64, String>,
    unknown_value: &str
) -> Result<String, Box<dyn Error>>

fn _register_as_special_value(
    token: &str,
    values: &HashMap<String, i64>,
    special_values: &mut HashMap<String, i64>
)

fn convert_tokens_to_ids(&self, tokens: Vec<&str>) -> Vec<i64>