pub struct Word2VecEmbeddingGenerator { /* private fields */ }Expand description
Word2Vec embedding generator
Implementations§
Source§impl Word2VecEmbeddingGenerator
impl Word2VecEmbeddingGenerator
Sourcepub fn new(
word2vec_config: Word2VecConfig,
embedding_config: EmbeddingConfig,
) -> Result<Self>
pub fn new( word2vec_config: Word2VecConfig, embedding_config: EmbeddingConfig, ) -> Result<Self>
Create a new Word2Vec embedding generator
Sourcepub fn load_model(&mut self, path: &str) -> Result<()>
pub fn load_model(&mut self, path: &str) -> Result<()>
Load pre-trained Word2Vec model
Sourcepub fn set_document_frequencies(&mut self, frequencies: HashMap<String, f32>)
pub fn set_document_frequencies(&mut self, frequencies: HashMap<String, f32>)
Set document frequencies for TF-IDF weighting
Sourcepub fn calculate_document_frequencies(
&mut self,
documents: &[String],
) -> Result<()>
pub fn calculate_document_frequencies( &mut self, documents: &[String], ) -> Result<()>
Calculate document frequencies from a corpus
Trait Implementations§
Source§impl AsAny for Word2VecEmbeddingGenerator
impl AsAny for Word2VecEmbeddingGenerator
Source§impl EmbeddingGenerator for Word2VecEmbeddingGenerator
impl EmbeddingGenerator for Word2VecEmbeddingGenerator
Source§fn generate(&self, content: &EmbeddableContent) -> Result<Vector>
fn generate(&self, content: &EmbeddableContent) -> Result<Vector>
Generate embedding for content
Source§fn generate_batch(&self, contents: &[EmbeddableContent]) -> Result<Vec<Vector>>
fn generate_batch(&self, contents: &[EmbeddableContent]) -> Result<Vec<Vector>>
Generate embeddings for multiple contents in batch
Source§fn dimensions(&self) -> usize
fn dimensions(&self) -> usize
Get the embedding dimensions
Source§fn config(&self) -> &EmbeddingConfig
fn config(&self) -> &EmbeddingConfig
Get the model configuration
Auto Trait Implementations§
impl Freeze for Word2VecEmbeddingGenerator
impl RefUnwindSafe for Word2VecEmbeddingGenerator
impl Send for Word2VecEmbeddingGenerator
impl Sync for Word2VecEmbeddingGenerator
impl Unpin for Word2VecEmbeddingGenerator
impl UnwindSafe for Word2VecEmbeddingGenerator
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
Mutably borrows from an owned value. Read more
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>
Converts
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>
Converts
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 moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<T> PolicyExt for Twhere
T: ?Sized,
impl<T> PolicyExt for Twhere
T: ?Sized,
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.