Skip to main content

Module config

Module config 

Source

Structs§

DataLoader
Utility for batching and streaming sentence data.
EarlyStoppingConfig
Configuration for early stopping during training.
TrainingConfig
Training hyperparameters for embedding models.
TrainingData
Container for tokenized sentences and the vocabulary mapping.

Enums§

LearningRateSchedule
Learning rate schedule variants.
ModelType
Supported embedding model architectures.

Constants§

DEFAULT_BATCH_SIZE
Default mini-batch size.
DEFAULT_CONTEXT_WINDOW
Default context window size.
DEFAULT_EMBEDDING_DIM
Default embedding dimension.
DEFAULT_EPOCHS
Default number of training epochs.
DEFAULT_LEARNING_RATE
Default learning rate.
DEFAULT_NEGATIVE_SAMPLES
Default number of negative samples.
DEFAULT_VALIDATION_RATIO
Default validation ratio (0.0 = no validation).