Struct PyTorchEmbedder 

pub struct PyTorchEmbedder { /* private fields */ }

PyTorch model wrapper for embedding generation

Implementations

impl PyTorchEmbedder

pub fn new(config: PyTorchConfig) -> Result<Self>

Create a new PyTorch embedder

pub fn load_model(&mut self) -> Result<()>

Load PyTorch model from file

pub fn embed_text(&self, text: &str) -> Result<Vector>

Generate embeddings for text

pub fn embed_batch(&self, texts: &[String]) -> Result<Vec<Vector>>

Generate embeddings for multiple texts

pub fn get_metadata(&self) -> Option<&PyTorchModelMetadata>

Get model metadata

pub fn get_dimensions(&self) -> Option<usize>

Get embedding dimensions

pub fn set_tokenizer(&mut self, tokenizer: PyTorchTokenizer)

Update tokenizer

pub fn supports_mixed_precision(&self) -> bool

Check if model supports mixed precision

pub fn get_device(&self) -> &PyTorchDevice

Get current device

Trait Implementations

impl Debug for PyTorchEmbedder

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl EmbeddingGenerator for PyTorchEmbedder

fn generate_embedding(&self, content: &ContentItem) -> Result<Vector>

Generate embedding vector from content

fn generate_batch_embeddings( &self, content: &[ContentItem], ) -> Result<Vec<ProcessingResult>>

Generate embeddings for a batch of content items

fn embedding_dimensions(&self) -> usize

Get the embedding dimensions

fn get_config(&self) -> Value

Get generator configuration

fn is_ready(&self) -> bool

Check if the generator is ready

fn get_statistics(&self) -> GeneratorStatistics

Get generator statistics