Struct TensorFlowEmbedder 

pub struct TensorFlowEmbedder { /* private fields */ }

TensorFlow embedder for generating embeddings

Implementations

impl TensorFlowEmbedder

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

Create a new TensorFlow embedder

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

Load and initialize the TensorFlow model

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

Generate embeddings for text content

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

Generate embeddings for multiple texts

pub fn get_model_info(&self) -> Option<&TensorFlowModelInfo>

Get model information

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

Get output dimensions

pub fn set_preprocessing_pipeline(&mut self, pipeline: PreprocessingPipeline)

Update preprocessing pipeline

Trait Implementations

impl Debug for TensorFlowEmbedder

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

Formats the value using the given formatter.

impl EmbeddingGenerator for TensorFlowEmbedder

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