Struct Word2Vec 

pub struct Word2Vec {
    pub embedding_dim: usize,
    pub window_size: usize,
    pub min_count: usize,
    pub learning_rate: f32,
    pub epochs: usize,
    /* private fields */
}

Word2Vec Skip-gram model (simplified)

Fields

embedding_dim: usize

window_size: usize

min_count: usize

learning_rate: f32

epochs: usize

Implementations

impl Word2Vec

pub fn new(embedding_dim: usize) -> Self

pub fn window_size(self, size: usize) -> Self

pub fn min_count(self, count: usize) -> Self

pub fn fit(&mut self, texts: &[String])

pub fn get_vector(&self, word: &str) -> Option<&[f32]>

pub fn similarity(&self, word1: &str, word2: &str) -> Option<f32>

pub fn vocab_size(&self) -> usize