Trait SentenceEmbedder

pub trait SentenceEmbedder: Sized {
    // Required methods
    fn embedding_size(&self) -> usize;
    fn fit<S>(self, sentences: &[S]) -> Result<Self>
       where S: AsRef<str>;
    fn embeddings<I, S>(&self, sentences: I) -> Result<Array2<Float>>
       where I: IntoIterator<Item = S>,
             S: AsRef<str>;
}

Common behavior of our models for sentence embeddings.

Required Methods

fn embedding_size(&self) -> usize

Returns the number of dimensions for sentence embeddings.

fn fit<S>(self, sentences: &[S]) -> Result<Self>

where S: AsRef<str>,

Fits the model with input sentences.

fn embeddings<I, S>(&self, sentences: I) -> Result<Array2<Float>>

where I: IntoIterator<Item = S>, S: AsRef<str>,

Computes embeddings for input sentences using the fitted model.

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors

impl<W, P> SentenceEmbedder for Sif<'_, '_, W, P>

where W: WordEmbeddings, P: WordProbabilities,

impl<W, P> SentenceEmbedder for USif<'_, '_, W, P>

where W: WordEmbeddings, P: WordProbabilities,