Struct Word2VecBuilder 

pub struct Word2VecBuilder { /* private fields */ }

Builder for Word2Vec model

Implementations

impl Word2VecBuilder

pub fn new() -> Self

Create a new builder with default configuration

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

Set vector size (default: 100)

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

Set window size (default: 5)

pub fn negative_samples(self, n: usize) -> Self

Set number of negative samples (default: 5)

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

Set minimum word count (default: 10)

pub fn sample(self, threshold: f64) -> Self

Set subsampling threshold (default: 1e-4)

pub fn alpha(self, alpha: f32) -> Self

Set initial learning rate (default: 0.025)

pub fn min_alpha(self, alpha: f32) -> Self

Set minimum learning rate (default: 0.0001)

pub fn epochs(self, epochs: usize) -> Self

Set number of epochs (default: 3)

pub fn threads(self, threads: usize) -> Self

Set number of threads (default: 8)

pub fn build_from_corpus<P: AsRef<Path>>( self, corpus_path: P, ) -> Result<Word2Vec>

Build vocabulary from corpus and create model

Trait Implementations

impl Default for Word2VecBuilder

fn default() -> Word2VecBuilder

Returns the “default value” for a type.