Skip to main content

NormalizedEmbeddingModel

Struct NormalizedEmbeddingModel 

Source
pub struct NormalizedEmbeddingModel<M>
where M: EmbeddingModel,
{ /* private fields */ }
Expand description

归一化 Embedding 模型适配器

包装一个内部 EmbeddingModel,对输出向量进行 L2 归一化, 确保所有向量都是单位向量。适用于需要使用余弦相似度的场景。

Implementations§

Source§

impl<M> NormalizedEmbeddingModel<M>
where M: EmbeddingModel,

Source

pub fn new(inner: M) -> Self

创建归一化适配器

Source

pub fn l2_normalize(vector: &mut [f32])

对向量进行 L2 归一化

Source

pub fn inner(&self) -> &M

获取内部模型引用

Trait Implementations§

Source§

impl<M> EmbeddingModel for NormalizedEmbeddingModel<M>
where M: EmbeddingModel,

Source§

fn embed<'life0, 'life1, 'async_trait>( &'life0 self, text: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<Vec<f32>, AiError>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn embed_batch<'life0, 'life1, 'async_trait>( &'life0 self, texts: &'life1 [String], ) -> Pin<Box<dyn Future<Output = Result<Vec<Vec<f32>>, AiError>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn dimension(&self) -> usize

Source§

fn model_name(&self) -> &str

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.