Struct CodebookTrainer 

pub struct CodebookTrainer { /* private fields */ }

Codebook trainer using K-means clustering

Implementations

impl CodebookTrainer

pub fn new(config: TrainingConfig) -> Self

Create a new trainer

pub fn train( &self, codebook_config: CodebookConfig, data: &[f32], ) -> Result<Codebook>

Train a codebook on data

pub fn train_all( &self, model_id: &str, data_by_layer: &[(LayerType, Vec<f32>)], ) -> Result<LayerCodebook>

Train codebooks for all layer types