Module config 

Hyperparameter configuration for Word2Vec training.

Example

use word2vec::{Config, ModelType};

let cfg = Config {
    embedding_dim: 128,
    window_size: 5,
    negative_samples: 10,
    epochs: 10,
    learning_rate: 0.025,
    min_learning_rate: 0.0001,
    min_count: 5,
    subsample_threshold: 1e-3,
    model: ModelType::SkipGram,
    num_threads: 4,
    seed: 42,
};

assert_eq!(cfg.embedding_dim, 128);

Structs

Config

Full training configuration.

Enums

ModelType