Struct StaticModel 

pub struct StaticModel { /* private fields */ }

Static embedding model for Model2Vec

Implementations

impl StaticModel

pub fn from_pretrained<P: AsRef<Path>>( repo_or_path: P, token: Option<&str>, normalize: Option<bool>, subfolder: Option<&str>, ) -> Result<Self>

Load a Model2Vec model from a local folder or the HuggingFace Hub.

Arguments

• repo_or_path - HuggingFace repo ID or local path to the model folder.
• token - Optional HuggingFace token for authenticated downloads.
• normalize - Optional flag to normalize embeddings (default from config.json).
• subfolder - Optional subfolder within the repo or path to look for model files.

pub fn encode_with_args( &self, sentences: &[String], max_length: Option<usize>, batch_size: usize, ) -> Vec<Vec<f32>>

Encode texts into embeddings.

Arguments

• sentences - the list of sentences to encode.
• max_length - max tokens per text.
• batch_size - number of texts per batch.

pub fn encode(&self, sentences: &[String]) -> Vec<Vec<f32>>

Default encode: max_length=512, batch_size=1024

pub fn encode_single(&self, sentence: &str) -> Vec<f32>

Trait Implementations

impl Clone for StaticModel

fn clone(&self) -> StaticModel

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for StaticModel

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

Formats the value using the given formatter.