Trait finalfusion::similarity::WordSimilarityBy

pub trait WordSimilarityBy {
    fn word_similarity_by<F>(
        &self, 
        word: &str, 
        limit: usize, 
        similarity: F
    ) -> Option<Vec<WordSimilarityResult>>
    where
        F: FnMut(ArrayView2<f32>, ArrayView1<f32>) -> Array1<f32>;
}

Trait for word similarity queries with a custom similarity function.

Required methods

fn word_similarity_by<F>(
    &self,
    word: &str,
    limit: usize,
    similarity: F
) -> Option<Vec<WordSimilarityResult>> where
    F: FnMut(ArrayView2<f32>, ArrayView1<f32>) -> Array1<f32>, 

Find words that are similar to the query word using the given similarity function.

The similarity function should return, given the embeddings matrix and the word vector a vector of similarity scores. At most, limit results are returned.

Loading content...

Implementors

impl<V, S> WordSimilarityBy for Embeddings<V, S> where
    V: Vocab,
    S: StorageView, 

fn word_similarity_by<F>(
    &self,
    word: &str,
    limit: usize,
    similarity: F
) -> Option<Vec<WordSimilarityResult>> where
    F: FnMut(ArrayView2<f32>, ArrayView1<f32>) -> Array1<f32>, 

Loading content...