Re-exports§
pub use io::SentenceIterator;pub use io::WriteModelBinary;pub use io::WriteModelText;pub use io::WriteModelWord2Vec;
Modules§
Structs§
- Bucket
Config - Hyperparameters for bucket-vocabs.
- Common
Config - Common embedding model hyperparameters.
- Counted
Type - Depembeds
Config - Hyperparameters for Dependency Embeddings.
- Depembeds
Trainer - Dependency embeddings Trainer.
- Dependency
Iterator - Struct to iterate over the dependencies in a
conllu::DepGraph. - NGram
Config - Hyperparameters for ngram-vocabs.
- SGD
- Stochastic gradient descent
- Simple
Vocab - Generic corpus vocabulary type.
- Simple
Vocab Config - Hyperparameters for simple vocabs.
- Skip
Gram Config - Hyperparameters for SkipGram-like models.
- Skipgram
Trainer - Skipgram Trainer
- Subword
Vocab - A corpus vocabulary with subword lookup.
- Subword
Vocab Config - Hyperparameters for Subword vocabs.
- Train
Model - Training model.
- Vocab
Builder - Generic builder struct to count types.
Enums§
- Bucket
Indexer Type - Bucket Indexer Types
- Cutoff
- Cutoff to determine vocabulary size.
- Dependency
- Enum for different types of dependencies. Typed through direction, depth, attached form and label.
- Loss
Type - Losses.
- Model
Type - Model types.
Traits§
- DepIter
- Trait offering adapters for
DependencyIterator. - Trainer
- Trainer Trait.
- Vocab
- Trait for lookup of indices.